KeyError: 'depth_actor_state_dict'

[XanderWangXY]

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]

I have solved it..

[sdalal1]

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

[XanderWangXY]

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

[KUsaka-xu]

Have you deploy the model into real robot?