浏览代码

Fix not saving .nn file after max_timesteps (#1896)

Sends close command when closing workers.
/develop-generalizationTraining-TrainerController
Jonathan Harper 5 年前
当前提交
8616db5d
共有 1 个文件被更改,包括 1 次插入1 次删除
  1. 2
      ml-agents-envs/mlagents/envs/subprocess_environment.py

2
ml-agents-envs/mlagents/envs/subprocess_environment.py


return response
def close(self):
self.conn.send(EnvironmentCommand("close"))
self.process.join()

elif cmd.name == 'global_done':
_send_response('global_done', env.global_done)
elif cmd.name == 'close':
env.close()
break
except KeyboardInterrupt:
print('UnityEnvironment worker: keyboard interrupt')

正在加载...
取消
保存