search

stepjam / RLBench

A large-scale benchmark and learning environment.
https://sites.google.com/corp/view/rlbench
Other
1.16k stars 235 forks source link

Imitation learning prediction - Question #49

Closed raghada closed 4 years ago

raghada commented 4 years ago

For the imitation learning example script, is it possible to send the predicted actions to the scene instead of just computing the loss?

By this I mean, can I test the robot with my networks' prediction?

I basically know how to do this with PyRep, but not with RLBench. Would really appreciate it if you can advise.

Thanks for the great work and effort you are putting on this!

stepjam commented 4 years ago

Something like this? https://github.com/stepjam/RLBench/blob/master/examples/single_task_rl.py

raghada commented 4 years ago

Yes, very much appreciated :)