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
|
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):
|
def predict_single(input, target, model):
|
||||||
inputs = input.unsqueeze(0)
|
inputs = input.unsqueeze(0)
|
||||||
|
|
Loading…
Reference in New Issue