google-research / ravens

Train robotic agents to learn pick and place with deep learning for vision-based manipulation in PyBullet. Transporter Nets, CoRL 2020.
https://transporternets.github.io
Apache License 2.0
538 stars 93 forks source link

Six DOF training error #10

Open Codie-land opened 3 years ago

Codie-land commented 3 years ago

Hello, first of all thank you for making this repository open source. I tried replicating the results given in your paper with the block insertion 6 degrees of freedom task. However, I am unsure which model and agent was used to train it in the paper, because I am getting errors when I try to use the transporter_6d agent. I have attached the traceback call:

image

I hope you can help me train the block insertion with 6dof task. Not sure if I am using the right agent for it. Any help is appreciated, thank you.