added demo tfx images

2019-06-29 18:42:39 -04:00 · 2019-06-29 18:42:39 -04:00 · 1460987b40
parent f39f8ed1a9
commit 1460987b40
1 changed files with 1 additions and 0 deletions
--- a/examples/new_project_templates/trainer_gpu_cluster_template.py
+++ b/examples/new_project_templates/trainer_gpu_cluster_template.py
@ -56,6 +56,7 @@ def main(hparams, cluster, results_dict):

    # init experiment
    log_dir = os.path.dirname(os.path.realpath(__file__))
+    log_dir = os.path.join(log_dir, 'test_tube_demo_logs')
    exp = Experiment(
        name='test_tube_exp',
        save_dir=log_dir,