minor adjustment

uoguelph-mlrg · Aug 12, 2017 · 4bf0ebc · 4bf0ebc
1 parent 56e93ae
commit 4bf0ebc
Show file tree

Hide file tree

Showing 2 changed files with 1 addition and 3 deletions.
diff --git a/theanompi/gosgd_worker.py b/theanompi/gosgd_worker.py
@@ -59,8 +59,6 @@ def run(self, model):
             recorder.start_epoch()
 
             # train
-
-            print(model.data.n_batch_train)
 
             for batch_i in range(model.data.n_batch_train):
 

diff --git a/theanompi/models/__init__.py b/theanompi/models/__init__.py
@@ -1,4 +1,4 @@
 from theanompi.models.alex_net import AlexNet
-from theanompi.models.alex_net_sc import AlexNet_sc
+# from theanompi.models.alex_net_sc import AlexNet_sc
 from theanompi.models.cifar10 import Cifar10_model
 from theanompi.models.googlenet import GoogLeNet