From 638e8e92ea6c0592ec81e1a18bbda954b2adef48 Mon Sep 17 00:00:00 2001 From: Alagris Date: Mon, 10 May 2021 12:00:34 +0200 Subject: [PATCH] no mongo --- train_model.py | 12 ++++++------ 1 file changed, 6 insertions(+), 6 deletions(-) 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')