From 1f5e8f084c9c322f642d3ea10d74de5ee442f9a8 Mon Sep 17 00:00:00 2001 From: Adrian Charkiewicz Date: Sat, 7 May 2022 21:55:55 +0200 Subject: [PATCH] epochs?? --- pytorch/pytorch.py | 19 ------------------- 1 file changed, 19 deletions(-) diff --git a/pytorch/pytorch.py b/pytorch/pytorch.py index 6ef38f1..650d030 100644 --- a/pytorch/pytorch.py +++ b/pytorch/pytorch.py @@ -166,25 +166,6 @@ def fit(epochs, lr, model, train_loader, val_loader, _run, opt_func=torch.optim. return history -# In[12]: - -try: - numberOfEpochParam = int(sys.argv[1]) -except: - # dafault val - numberOfEpochParam = 1500 - - -@ex.config -def my_config(): - epochs = numberOfEpochParam - - - - - -# In[27]: - def predict_single(input, target, model): inputs = input.unsqueeze(0)