take parameter from cli
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
Some checks failed
s444354-training/pipeline/head There was a failure building this commit
This commit is contained in:
parent
b26ec30db4
commit
4ee64747f8
@ -29,16 +29,18 @@ ex.observers.append(FileStorageObserver('my_runs'))
|
|||||||
|
|
||||||
# ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', db_name='sacred'))
|
# ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', db_name='sacred'))
|
||||||
|
|
||||||
#try:
|
|
||||||
# numberOfEpochParam = int(sys.argv[1])
|
|
||||||
#except:
|
|
||||||
# dafault val
|
try:
|
||||||
#numberOfEpochParam = 1500
|
numberOfEpochParam = int(sys.argv[1])
|
||||||
|
except:
|
||||||
|
numberOfEpochParam = 1500
|
||||||
|
|
||||||
@ex.config
|
@ex.config
|
||||||
def my_config():
|
def my_config():
|
||||||
global epochs
|
global epochs
|
||||||
epochs = 1500
|
epochs = numberOfEpochParam
|
||||||
|
|
||||||
|
|
||||||
|
|
||||||
@ -200,8 +202,8 @@ with open("result.txt", "w+") as file:
|
|||||||
@ex.automain
|
@ex.automain
|
||||||
def main(epochs, _run):
|
def main(epochs, _run):
|
||||||
lr = 1e-6
|
lr = 1e-6
|
||||||
my_config()
|
#my_config()
|
||||||
print("number of epochs is: ", epochs)
|
#print("number of epochs is: ", epochs)
|
||||||
history5 = fit(epochs, lr, model, train_loader, val_loader)
|
history5 = fit(epochs, lr, model, train_loader, val_loader)
|
||||||
|
|
||||||
#ex.run()
|
#ex.run()
|
||||||
|
Loading…
Reference in New Issue
Block a user