UT-Austin-RPL / VIOLA

Official implementation for VIOLA
MIT License
107 stars 7 forks source link

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

Open BaymaxShao opened 1 year ago

BaymaxShao commented 1 year ago

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 commented 11 months ago

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 commented 9 months ago

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

HeegerGao commented 4 months ago

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

pip install robomimic==0.3.0