--play now works with LSTM (#595)
This commit is contained in:
@@ -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
|
||||
|
Reference in New Issue
Block a user