search

jyf588 / transformer-inertial-poser

Python implementation accompanying the Transformer Inertial Poser paper at SIGGRAPH Asia 2022
https://arxiv.org/abs/2203.15720
Other
71 stars 11 forks source link

The problem about viz_generated_sbp.py #18

Open huzijun1996 opened 6 months ago

huzijun1996 commented 6 months ago

Traceback (most recent call last): File "viz_generated_sbp.py", line 22, in import sim_agent ModuleNotFoundError: No module named 'sim_agent'

Traceback (most recent call last): File "viz_generated_sbp.py", line 31, in from transformers.utils import set_seed ImportError: cannot import name 'set_seed' from 'transformers.utils' (/home/xxx/anaconda3/envs/tip22/lib/python3.8/site-packages/transformers/utils/init.py)

I cannot find sim_agent. I cannot find set_seed from transformers.util. What should I do?

amaliacontiero commented 6 months ago

As mentioned by the author, viz_generated_sbp.py is an uncleaned script and you need to modify it to get it to work.

For starters, these two errors can be fixed by replacing line 22 with "import bullet_agent" and line 31 with "from learning_utils import set_seed". Look at the rest of the python files for useful tools and the commented out code and args of viz_generated_sbp.py to get your desired result. You need to edit accordingly.

huzijun1996 commented 6 months ago

Thank you for your answer. My problem has been resolved.