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

add int
All checks were successful
s444354-training/pipeline/head This commit looks good
Details
s444354-evaluation/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-07 12:30:24 +02:00
parent 3033b4db86
commit 87fe38d3d1
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

@ -149,7 +149,7 @@ def fit(epochs, lr, model, train_loader, val_loader, opt_func=torch.optim.SGD):
# In[12]:
epochs = sys.argv[1]
epochs = int(sys.argv[1])
lr = 1e-6
history5 = fit(epochs, lr, model, train_loader, val_loader)