fix train_model.py
@@ -227,10 +227,10 @@ class TrainLoop():
# reload best validation model
self.agent = create_agent(opt)
- # _rep, wrld = run_eval(self.agent, opt, 'valid', write_log=True)
- # wrld.shutdown() # may need to shut down threads, remote connections
- # _rep, wrld = run_eval(self.agent, opt, 'test', write_log=True)
- # wrld.shutdown() # may need to shut down threads, remote connections
+ _rep, wrld = run_eval(self.agent, opt, 'valid', write_log=True)
+ wrld.shutdown() # may need to shut down threads, remote connections
+ _rep, wrld = run_eval(self.agent, opt, 'test', write_log=True)
+ wrld.shutdown() # may need to shut down threads, remote connections
if __name__ == '__main__':
