s444354/ium_444354
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

changed order
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-07 21:32:11 +02:00
parent 555fb25ac9
commit fb3ea65baa
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch/pytorch.py
View File

@ -132,7 +132,7 @@ def evaluate(model, val_loader):
return model.validation_epoch_end(outputs) return model.validation_epoch_end(outputs)
@ex.capture @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 = [] history = []
optimizer = opt_func(model.parameters(), lr) optimizer = opt_func(model.parameters(), lr)
for epoch in range(epochs): for epoch in range(epochs):