From 87fe38d3d1cb1223854d9590ad4aa6da514d5f6d Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sat, 7 May 2022 12:30:24 +0200 Subject: [PATCH] add int --- pytorch/pytorch.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index bde8aab..4d89412 100644 --- 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)