Commit 3796ab01 authored by jyotish's avatar jyotish
Browse files

Update rollout.py

parent ee841c68
......@@ -20,6 +20,7 @@ def main():
episode_count = 0
while episode_count < 20:
_, _, done, _ = env.step(env.action_space.sample())
print("Completed a step")
if done:
episode_count += 1
print(episode_count)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment