Zaktualizuj 'train-mlflow.py'

2022-05-15 13:05:15 +02:00 · 2022-05-15 13:05:15 +02:00 · 7c45c9b09b
commit 7c45c9b09b
parent ac9eca3ddd
1 changed files with 1 additions and 1 deletions
--- a/train-mlflow.py
+++ b/train-mlflow.py
@ -75,7 +75,7 @@ model = Model()
 criterion = nn.BCELoss()
 optimizer = torch.optim.SGD(model.parameters(), lr=0.01)

-#epochs = int(sys.argv[1])
+epochs = EPOCHS

 def my_main(epochs):
 # Trening