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
Block a user