fixed new generator

2026-09-09 11:13:26 +08:00 · 2019-11-18 12:20:27 -07:00
parent fbd6a58ae0
commit dda5ae9b2c
1 changed files with 1 additions and 1 deletions
@@ -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