shellmiao 11 месяцев назад
Родитель
Сommit
e98ce2b751
1 измененных файлов с 1 добавлено и 1 удалено
  1. 1 1
      applications/fedssl/client_with_pgfed.py

+ 1 - 1
applications/fedssl/client_with_pgfed.py

@@ -95,7 +95,7 @@ class FedSSLWithPgFedClient(FedSSLClient):
         
         # delete later
         all_grads_none = True
-        for p in zip(self.model.parameters()):
+        for p in self.model.parameters():
             if p.grad is not None:
                 all_grads_none = False
         if all_grads_none: