File size: 407 Bytes
f3eaffd | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 | import gym
import rlbench.gym
env = gym.make('reach_target-state-v0', render_mode='human')
training_steps = 120
episode_length = 40
for i in range(training_steps):
if i % episode_length == 0:
print('Reset Episode')
obs = env.reset()
obs, reward, terminate, _ = env.step(env.action_space.sample())
env.render() # Note: rendering increases step time.
print('Done')
env.close()
|