From 90d950dc0d42ea912f31e7ba44deeefd62b23a9a Mon Sep 17 00:00:00 2001 From: Nick Walton Date: Mon, 18 Nov 2019 17:56:21 -0700 Subject: [PATCH] update --- console_play.py | 2 +- generator/gpt2/gpt2_generator.py | 6 +++--- 2 files changed, 4 insertions(+), 4 deletions(-) diff --git a/console_play.py b/console_play.py index 37eccf0..cbba257 100644 --- a/console_play.py +++ b/console_play.py @@ -46,7 +46,7 @@ def play_unconstrained(): action = remove_profanity(action) # action = first_to_second_person(action) - result = story_manager.act(action) + result = "\n" + story_manager.act(action) if player_died(result): console_print(result + "\nGAME OVER") break diff --git a/generator/gpt2/gpt2_generator.py b/generator/gpt2/gpt2_generator.py index 13456d4..234ba91 100644 --- a/generator/gpt2/gpt2_generator.py +++ b/generator/gpt2/gpt2_generator.py @@ -7,7 +7,7 @@ from generator.gpt2.src import sample, encoder, model import json import numpy as np -tf.logging.set_verbosity(tf.logging.ERROR) +tf.compat.v1.logging.set_verbosity(tf.compat.v1.logging.ERROR) class GPT2Generator: @@ -36,8 +36,8 @@ class GPT2Generator: self.sess = tf.compat.v1.Session(config=config) self.context = tf.placeholder(tf.int32, [self.batch_size, None]) - np.random.seed(seed) - tf.set_random_seed(seed) + #np.random.seed(seed) + # tf.set_random_seed(seed) self.output = sample.sample_sequence( hparams=hparams, length=self.generate_num, context=self.context,