diff --git a/.ipynb_checkpoints/write_poem-checkpoint.py b/.ipynb_checkpoints/write_poem-checkpoint.py index 797aad4..9bd6408 100644 --- a/.ipynb_checkpoints/write_poem-checkpoint.py +++ b/.ipynb_checkpoints/write_poem-checkpoint.py @@ -210,7 +210,7 @@ def start_model(): now = int(time.time()) - args = config_sample('--model_dir /home/jovyan/work/results/output_poem --length 16 --seed {}'.format(now)) + args = config_sample('--model_dir ./results/output_poem --length 16 --seed {}'.format(now)) writer = WritePoem(args) return writer diff --git a/write_poem.py b/write_poem.py index 797aad4..9bd6408 100644 --- a/write_poem.py +++ b/write_poem.py @@ -210,7 +210,7 @@ def start_model(): now = int(time.time()) - args = config_sample('--model_dir /home/jovyan/work/results/output_poem --length 16 --seed {}'.format(now)) + args = config_sample('--model_dir ./results/output_poem --length 16 --seed {}'.format(now)) writer = WritePoem(args) return writer