Six DOF training error

[Codie-land]

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:

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.