add int

2022-05-07 12:30:24 +02:00 · 2022-05-07 12:30:24 +02:00 · 87fe38d3d1
parent 3033b4db86
commit 87fe38d3d1
1 changed files with 1 additions and 1 deletions
--- a/pytorch/pytorch.py
+++ b/pytorch/pytorch.py
@ -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)