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

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

Browse Source
This commit is contained in:
Adrian Charkiewicz 2022-05-07 21:48:07 +02:00
parent 2a5e48f63c
commit 960d2ebcd3
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
pytorch/pytorch.py
View File

@ -29,15 +29,15 @@ 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: #try:
numberOfEpochParam = int(sys.argv[1]) # numberOfEpochParam = int(sys.argv[1])
except: #except:
# dafault val # dafault val
numberOfEpochParam = 1500 #numberOfEpochParam = 1500
@ex.config @ex.config
def my_config(): def my_config():
epochs = numberOfEpochParam epochs = 1500
@ -217,6 +217,7 @@ with open("result.txt", "w+") as file:
def main(): def main():
lr = 1e-6 lr = 1e-6
my_config() my_config()
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()