--play now works with LSTM (#595)

This commit is contained in:
R1ckF
2018-09-17 21:33:39 +00:00
committed by pzhokhov
parent 0c6f357936
commit 68a29d0ab3

View File

@@ -221,8 +221,11 @@ def main():
logger.log("Running trained model")
env = build_env(args)
obs = env.reset()
def initialize_placeholders(nlstm=128,**kwargs):
return np.zeros((args.num_env, 2*nlstm)), np.zeros((1))
state, dones = initialize_placeholders(**extra_args)
while True:
actions = model.step(obs)[0]
actions, _, state, _ = model.step(obs,S=state, M=dones)
obs, _, done, _ = env.step(actions)
env.render()
done = done.any() if isinstance(done, np.ndarray) else done