epochs??
s444354-training/pipeline/head There was a failure building this commit
Details
s444354-training/pipeline/head There was a failure building this commit
Details
This commit is contained in:
parent
960d2ebcd3
commit
1f5e8f084c
|
@ -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)
|
||||
|
|
Loading…
Reference in New Issue