fix: test

applications/fedssl/client_with_pgfed.py

@@ -61,6 +61,7 @@ class FedSSLWithPgFedClient(FedSSLClient):
             data_count = 0 # delete later
             batch_loss = []
             for (batched_x1, batched_x2), _ in self.train_loader:
+                print(data_count)
                 if data_count >= 50:
                     break
                 x1, x2 = batched_x1.to(device), batched_x2.to(device)