From dda5ae9b2c99dfd8927bc6751b7a76d7044ddb75 Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 12:20:27 -0700 Subject: [PATCH] fixed new generator --- generator/gpt2/gpt2_generator.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 6348fca..947ad27 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -33,7 +33,7 @@ class GPT2Generator: hparams = model.default_hparams() with open(os.path.join(models_dir, self.model_name, 'hparams.json')) as f: hparams.override_from_dict(json.load(f)) - seed = 20 + seed = np.random.randint(0, 100000) config = tf.compat.v1.ConfigProto() config.gpu_options.allow_growth = True