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

mongo
Some checks failed
s434749-training/pipeline/head There was a failure building this commit
Details

Browse Source
This commit is contained in:
Alagris 2021-05-17 11:58:37 +02:00
parent 33491f6458
commit 3e0649786b
1 changed files with 5 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
train_model.py
View File

@ -22,11 +22,11 @@ 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@172.17.0.1: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')