From 3e0649786b758359a6f48db0ef26b520b7b3e117 Mon Sep 17 00:00:00 2001 From: Alagris Date: Mon, 17 May 2021 11:58:37 +0200 Subject: [PATCH] mongo --- train_model.py | 10 +++++----- 1 file changed, 5 insertions(+), 5 deletions(-) diff --git a/train_model.py b/train_model.py index 020c35c..0b2b8bf 100644 --- a/train_model.py +++ b/train_model.py @@ -22,11 +22,11 @@ import traceback ex = Experiment("CNN") ex.observers.append(FileStorageObserver('sacred_file_observer')) -# try: -# ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@localhost:27017', -# db_name='sacred')) -# except Exception as e: -# traceback.print_exc() +try: + ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2021@172.17.0.1:27017', + db_name='sacred')) +except Exception as e: + traceback.print_exc() device = torch.device('cuda:0') if torch.cuda.is_available() else torch.device('cpu')