diff --git a/train.py b/train.py index 21632c5..34250a3 100644 --- a/train.py +++ b/train.py @@ -2,7 +2,7 @@ from sacred import Experiment from sacred.observers import MongoObserver, FileStorageObserver import os -os.environ["SACRED_NO_GIT"] = "1" + ex = Experiment('s487187-training', interactive=True, save_git_info=False) ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017', db_name='sacred')) @@ -10,7 +10,7 @@ ex.observers.append(MongoObserver(url='mongodb://admin:IUM_2021@172.17.0.1:27017 @ex.config def my_config(): - data_file = 'data.csv' + data_file = 'data.csv' model_file = 'model.h5' epochs = 10 batch_size = 32