shellmiao преди 1 година
родител
ревизия
d07799a223
променени са 1 файла, в които са добавени 1 реда и са изтрити 1 реда
  1. 1 1
      applications/fedssl/client_with_pgfed.py

+ 1 - 1
applications/fedssl/client_with_pgfed.py

@@ -90,7 +90,7 @@ class FedSSLWithPgFedClient(FedSSLClient):
 
         self.loss_minus = 0.0
         test_num = 0
-        self.optimizer.zero_grad()
+        optimizer.zero_grad()
         for (batched_x1, batched_x2), _ in self.train_loader:
             x1, x2 = batched_x1.to(self.device), batched_x2.to(self.device)
             test_num += x1.size(0)