Cleanup

fra98 · Feb 13, 2022 · 622bc19 · 622bc19
1 parent 2c0e5ff
commit 622bc19
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 143 deletions.
diff --git a/main_centralized.py b/main_centralized.py
@@ -19,7 +19,7 @@
 STEP_SIZE = [5, 10, 15]   # How many epochs before decreasing learning rate (if using a step-down policy)
 GAMMA = 0.1               # Multiplicative factor for learning rate step-down
 
-NET = LeNet5_MOD_LIGHT_BN
+NET = LeNet5_MOD_BN
 
 
 def run_training(net, criterion, optimizer, scheduler=None):

diff --git a/main_colab_fedAVG.py b/main_colab_fedAVG.py
diff --git a/main_colab_fedOPT.py b/main_colab_fedOPT.py