s434749/ium_434749
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

no mongo
All checks were successful
s434749-training/pipeline/head This commit looks good
Details

Browse Source
This commit is contained in:
Alagris 2021-05-10 12:00:34 +02:00
parent 16db0c341e
commit 638e8e92ea
1 changed files with 6 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
train_model.py
View File

@ -22,12 +22,12 @@ import traceback
ex = Experiment("CNN") ex = Experiment("CNN")
ex.observers.append(FileStorageObserver('sacred_file_observer')) ex.observers.append(FileStorageObserver('sacred_file_observer'))
try: # try:
ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', # ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017',
db_name='sacred')) # db_name='sacred'))
except Exception as e: # except Exception as e:
traceback.print_exc() # traceback.print_exc()
device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu') device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')