From 26c0b469ebc67c8b9988d47cd4baae625138c528 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Christoffer=20M=C3=BCller=20Madsen?= Date: Tue, 22 May 2018 20:49:10 +0200 Subject: [PATCH] restore restore_model --- main.py | 2 ++ 1 file changed, 2 insertions(+) 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`