Zaktualizuj 'train-sacred.py'

2022-05-08 19:27:06 +02:00 · 2022-05-08 19:27:06 +02:00 · 51428cd44e
commit 51428cd44e
parent d6a01d0b4c
1 changed files with 1 additions and 1 deletions
--- a/train-sacred.py
+++ b/train-sacred.py
@ -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)