epochs?

pytorch/pytorch.py

@@ -29,15 +29,15 @@ ex.observers.append(FileStorageObserver('my_runs'))
 
 # ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', db_name='sacred'))
 
-try: 
-  numberOfEpochParam = int(sys.argv[1])
-except:
-  # dafault val
-  numberOfEpochParam = 1500
+#try: 
+#  numberOfEpochParam = int(sys.argv[1])
+#except:
+#   dafault val
+#numberOfEpochParam = 1500
 
 @ex.config
 def my_config():
-  epochs = numberOfEpochParam
+  epochs = 1500
 
 
 
@@ -217,6 +217,7 @@ with open("result.txt", "w+") as file:
 def main():
     lr = 1e-6
     my_config()
+    print("number of epochs is: ", epochs)
     history5 = fit(epochs, lr, model, train_loader, val_loader)
 
 ex.run()