diff --git a/train_model.py b/train_model.py index d92104f..020c35c 100644 --- a/train_model.py +++ b/train_model.py @@ -22,12 +22,12 @@ 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@localhost: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')