Issue with `generate_cand.py`

[OkayJeff5]

Observation:
The README says to run:

python convert.py
python generate_cand.py

However, when I run generate_cand.py, I get:

KeyError: 'mvel'

Looking at the .pkl files generated by convert.py, they only contain:

['qpos', 'jpos', 'mpos', 'torque', 'grf']

There is no mvel key.

Proposal:
In generate_cand.py, should:

nf = joblib.load(f'data/imdy_train/{item}')['mvel'].shape[0]

be changed to:

nf = joblib.load(f'data/imdy_train/{item}')['mpos'].shape[0]

Question:
Could you confirm if these changes are correct?

[wululu66]

您好 我遇到和您一样的问题，请问您解决了吗