Zaktualizuj 'train-mlflow.py'

2022-05-15 13:06:50 +02:00 · 2022-05-15 13:06:50 +02:00 · 9626387f5d
commit 9626387f5d
parent 7c45c9b09b
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 = EPOCHS
+epochs = int(sys.argv[1]) if len(sys.argv) > 1 else 20

 def my_main(epochs):
 # Trening