diff --git a/train.py b/train.py index 6c9e3e4..90eddff 100644 --- a/train.py +++ b/train.py @@ -69,3 +69,5 @@ def run_experiment(): accuracy = train_model() ex.log_scalar('accuracy', accuracy) ex.add_artifact('model.h5') + +ex.run() \ No newline at end of file