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

epochs??
s444354-training/pipeline/head There was a failure building this commit Details

This commit is contained in:
Adrian Charkiewicz 2022-05-07 21:55:55 +02:00
parent 960d2ebcd3
commit 1f5e8f084c
1 changed files with 0 additions and 19 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
pytorch/pytorch.py
View File

@ -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)