diff --git a/main.py b/main.py index 2c4a449..c9b8d73 100644 --- a/main.py +++ b/main.py @@ -209,6 +209,8 @@ if __name__ == "__main__": elif args.eval: network = Network(config, config['model']) + network.restore_model() + for i in range(int(config['repeat_eval'])): start_episode = network.episodes_trained # Evaluation measures are described in `config`