Merge pull request #78 from KittenHero/master

Fix generate cmd run name
minimaxir · Jun 29, 2019 · 5f09600 · 5f09600
2 parents 7c8e969 + d5f0109
commit 5f09600
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/gpt_2_simple/gpt_2.py b/gpt_2_simple/gpt_2.py
@@ -740,6 +740,7 @@ def cmd_generate(nfiles, nsamples, folder,
                     'gpt2_gentext_{:%Y%m%d_%H%M%S}.txt'.format(datetime.utcnow()))
 
         generate_to_file(sess,
+                         run_name=run_name,
                          destination_path=gen_file,
                          length=length,
                          temperature=temperature,