Zaktualizuj 'train-sacred.py'

train-sacred.py

@@ -23,7 +23,7 @@ ex = Experiment("s478815",save_git_info=False)
 ex.observers.append(FileStorageObserver('experiment'))
 
 #ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password@localhost:27017', db_name='sacred'))  
-EPOCHS = int(sys.argv[1])
+
 @ex.config
 def my_config():
     epochs = 1000
@@ -77,7 +77,7 @@ optimizer = torch.optim.SGD(model.parameters(), lr=0.01)
 @ex.automain
 def my_main(epochs):
 # Trening
-    epochs = EPOCHS
+    #epochs = EPOCHS
     for epochs in range(epochs):
         y_predicted = model(X_training)
         loss = criterion(y_predicted,y_training)