search

chengxuxin / extreme-parkour

Train your parkour robot in less than 20 hours.
https://extreme-parkour.github.io
Other
472 stars 85 forks source link

KeyError: 'depth_actor_state_dict' #24

Open XanderWangXY opened 5 months ago

XanderWangXY commented 5 months ago

When I use command "python save_jit.py --exptid xxx-xx" I meet this error.

Traceback (most recent call last): File "/home/adc/pycharm-community-2023.3.2/plugins/python-ce/helpers/pydev/pydevd.py", line 1534, in _exec pydev_imports.execfile(file, globals, locals) # execute the script File "/home/adc/pycharm-community-2023.3.2/plugins/python-ce/helpers/pydev/_pydev_imps/_pydev_execfile.py", line 18, in execfile exec(compile(contents+"\n", file, 'exec'), glob, loc) File "/home/adc/extreme-parkour/legged_gym/legged_gym/scripts/save_jit.py", line 123, in play(args) File "/home/adc/extreme-parkour/legged_gym/legged_gym/scripts/save_jit.py", line 91, in play policy.actor.load_state_dict(ac_state_dict['depth_actor_state_dict'], strict=True) KeyError: 'depth_actor_state_dict'

How can I solve it?

XanderWangXY commented 5 months ago

I have solved it..

sdalal1 commented 5 months ago

I am getting the same error. Can you share how you solved it?

XanderWangXY commented 5 months ago

I am getting the same error. Can you share how you solved it?

change "--exptid" value to the same file name as the trained distill model file python save_jit.py --exptid yyy-yy

Indiasuperman commented 5 days ago

Have you deploy the model into real robot?