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')