diff --git a/table/experiment.py b/table/experiment.py index 1c138a6..4c70782 100644 --- a/table/experiment.py +++ b/table/experiment.py @@ -990,13 +990,13 @@ def weight_samples(samples): f_train.close() if agent.model.get_global_step() >= FLAGS.n_steps: - if FLAGS.save_replay_buffer_at_end: - all_replay = os.path.join(get_experiment_dir(), - 'all_replay_samples_{}.txt'.format(self.name)) - with codecs.open(all_replay, 'w', encoding='utf-8') as f: - samples = replay_buffer.all_samples(envs, agent=None) - samples = [s for s in samples if not replay_buffer_copy.contain(s.traj)] - f.write(show_samples(samples, envs[0].de_vocab, None)) + if FLAGS.save_replay_buffer_at_end: + all_replay = os.path.join(get_experiment_dir(), + 'all_replay_samples_{}.txt'.format(self.name)) + with codecs.open(all_replay, 'w', encoding='utf-8') as f: + samples = replay_buffer.all_samples(envs, agent=None) + samples = [s for s in samples if not replay_buffer_copy.contain(s.traj)] + f.write(show_samples(samples, envs[0].de_vocab, None)) tf.logging.info('{} finished'.format(self.name)) return