restore restore_model

This commit is contained in:
Christoffer Müller Madsen 2018-05-22 20:49:10 +02:00
parent f170bad9b1
commit 26c0b469eb

View File

@ -209,6 +209,8 @@ if __name__ == "__main__":
elif args.eval: elif args.eval:
network = Network(config, config['model']) network = Network(config, config['model'])
network.restore_model()
for i in range(int(config['repeat_eval'])): for i in range(int(config['repeat_eval'])):
start_episode = network.episodes_trained start_episode = network.episodes_trained
# Evaluation measures are described in `config` # Evaluation measures are described in `config`