diff --git a/train_model.py b/train_model.py index c82006d..90ed416 100644 --- a/train_model.py +++ b/train_model.py @@ -49,8 +49,8 @@ def main(batch_size, epochs, _run): def setup_experiment(): ex = Experiment('Predict power output for a given time') ex.observers.append(FileStorageObserver('sacred_runs')) - # ex.observers.append(MongoObserver(url='mongodb://mongo_user:mongo_password_IUM_2022@localhost:27017', - # db_name='sacred')) + ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', + db_name='sacred')) return ex