From fb3ea65baa2f2d6fdb64db0b256c17aaec3110fe Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sat, 7 May 2022 21:32:11 +0200 Subject: [PATCH] changed order --- pytorch/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index 9d28a36..5fae4d1 100644 --- a/pytorch/pytorch.py +++ b/pytorch/pytorch.py @@ -132,7 +132,7 @@ def evaluate(model, val_loader): return model.validation_epoch_end(outputs) @ex.capture -def fit(epochs, lr, model, train_loader, val_loader, opt_func=torch.optim.SGD, _run): +def fit(epochs, lr, model, train_loader, val_loader,_run, opt_func=torch.optim.SGD): history = [] optimizer = opt_func(model.parameters(), lr) for epoch in range(epochs):