alexsax / midlevel-reps

Mid-Level Visual Representations Improve Generalization and Sample Efficiency for Learning Visuomotor Policies
http://perceptual.actor
MIT License
107 stars 19 forks source link

Error while importing environment #15

Open jsdd25 opened 3 years ago

jsdd25 commented 3 years ago

Hi, Thanks for providing this tool. I have installed the latest habitat-lab and habitat-sim. Then, I want to run the experiment, but I get an error when importing the environment:

jin@jin-MS-7B94:~/RL-code/midlevel-reps$ python -m scripts.train_rl midlevel_logs run_training with uuid=normals cfg_habitat taskonomy_decoding  cfg.saving.log_interval=10 cfg.env.num_processes=2 cfg.env.num_val_processes=1
Traceback (most recent call last):
  File "/home/jin/anaconda3/envs/habitat7/lib/python3.6/runpy.py", line 193, in _run_module_as_main
    "__main__", mod_spec)
  File "/home/jin/anaconda3/envs/habitat7/lib/python3.6/runpy.py", line 85, in _run_code
    exec(code, run_globals)
  File "/home/jin/RL-code/midlevel-reps/scripts/train_rl.py", line 30, in <module>
    from evkit.env.wrappers import ProcessObservationWrapper
  File "/home/jin/RL-code/midlevel-reps/evkit/env/__init__.py", line 3, in <module>
    from .distributed_factory import DistributedEnv
  File "/home/jin/RL-code/midlevel-reps/evkit/env/distributed_factory.py", line 1, in <module>
    from evkit.env.util.vec_env.subproc_vec_embodied_env import SubprocVecEmbodiedEnv
  File "/home/jin/RL-code/midlevel-reps/evkit/env/util/vec_env/subproc_vec_embodied_env.py", line 3, in <module>
    from baselines.common.vec_env import VecEnv, CloudpickleWrapper
ModuleNotFoundError: No module named 'baselines'

Could you give me some help? Thank you very much!