eval.mp4 works!

This commit is contained in:
Cadene
2024-01-30 23:30:14 +00:00
parent 1144819c29
commit 1e52499490
4 changed files with 127 additions and 15 deletions

View File

@@ -31,6 +31,8 @@ def test_simxarm(task, from_pixels, pixels_only):
print("observation_spec:", env.observation_spec)
print("action_spec:", env.action_spec)
print("reward_spec:", env.reward_spec)
print("done_spec:", env.done_spec)
print("success_spec:", env.success_spec)
td = env.reset()
print("reset tensordict", td)