Cannot import name 'CropRandomizer' from 'robomimic.models.base_nets'

[BaymaxShao]

I try training the model with your generated datasets, and get this error:
Cannot import name 'CropRandomizer' from 'robomimic.models.base_nets'

With suspecting incorrect installation of robomimic, I go to https://github.com/ARISE-Initiative/robomimic/blob/master/robomimic/models/base_nets.py to check. However, I can't find CropRandomizer and Randomizer either.

Could you help me figure out why and how?

[zhuyifengzju]

Hi, it could be that they've updated their codebase. Please modify the code accordingly to accommodate their new changes, or you will need to install an older version of robomimic to use these classes.

[ramkumarkoppu]

having pip installation from requirements.txt file. what is the python version proven to be working for this repo?

[HeegerGao]

Encounter the same issue in another project. Upgrading the robomimic version to 0.3.0 solves this problem.

pip install robomimic==0.3.0