restore restore_model

2018-05-22 20:49:10 +02:00 · 2018-05-22 20:49:10 +02:00 · 26c0b469eb
commit 26c0b469eb
parent f170bad9b1
1 changed files with 2 additions and 0 deletions
--- 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`