update generator

2026-09-09 11:13:26 +08:00 · 2019-11-18 12:10:45 -07:00
parent 58b859637f
commit d3bdb6e3ac
2 changed files with 2 additions and 1 deletions
@@ -35,7 +35,8 @@ class GPT2Generator:
            hparams.override_from_dict(json.load(f))
        seed = 20

-        self.sess = tf.Session(graph=tf.Graph())
+        graph = tf.Graph()
+        self.sess = tf.Session(graph=graph)
        self.context = tf.placeholder(tf.int32, [self.batch_size, None])
        np.random.seed(seed)
        tf.set_random_seed(seed)