From 3e9f37a382ee1739d09306b929f04ebc7d75c479 Mon Sep 17 00:00:00 2001 From: William Falcon Date: Wed, 3 Apr 2019 09:07:20 -0400 Subject: [PATCH] fixes #4 --- demo/__init__.py | 0 demo/fully_featured_trainer.py | 13 +++++++------ 2 files changed, 7 insertions(+), 6 deletions(-) create mode 100644 demo/__init__.py diff --git a/demo/__init__.py b/demo/__init__.py new file mode 100644 index 0000000000..e69de29bb2 diff --git a/demo/fully_featured_trainer.py b/demo/fully_featured_trainer.py index fc4b16d008..f971f28300 100644 --- a/demo/fully_featured_trainer.py +++ b/demo/fully_featured_trainer.py @@ -17,7 +17,7 @@ np.random.seed(SEED) # --------------------- # DEFINE MODEL HERE # --------------------- -from demo.example_model import ExampleModel +from example_model import ExampleModel # --------------------- AVAILABLE_MODELS = { @@ -57,13 +57,14 @@ def main(hparams, cluster, results_dict): sleep(process_position + 1) # init experiment + log_dir = os.path.dirname(os.path.realpath(__file__)) exp = Experiment( - name=hparams.tt_name, - debug=hparams.debug, - save_dir=hparams.tt_save_path, - version=hparams.hpc_exp_number, + name='test_tube_exp', + debug=True, + save_dir=log_dir, + version=0, autosave=False, - description=hparams.tt_description + description='test demo' ) exp.argparse(hparams)