restore restore_model
This commit is contained in:
parent
f170bad9b1
commit
26c0b469eb
2
main.py
2
main.py
|
@ -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`
|
||||||
|
|
Loading…
Reference in New Issue
Block a user