Update experiments.py

Xtra-Computing · Aug 17, 2022 · 66095cc · 66095cc
1 parent 5371adb
commit 66095cc
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/experiments.py b/experiments.py
@@ -670,7 +670,7 @@ def local_train_net_scaffold(nets, selected, global_model, c_nets, c_global, arg
         logger.info("net %d final test acc %f" % (net_id, testacc))
         avg_acc += testacc
     for key in total_delta:
-        total_delta[key] /= len(selected)
+        total_delta[key] /= args.n_parties
     c_global_para = c_global.state_dict()
     for key in c_global_para:
         if c_global_para[key].type() == 'torch.LongTensor':