search

ac-93 / tactile_gym

Suite of PyBullet reinforcement learning environments targeted towards using tactile data as the main form of observation.
GNU General Public License v3.0
118 stars 20 forks source link

Cannot find ImpalaCNN #3

Closed eanswer closed 2 years ago

eanswer commented 2 years ago

When I try to run train_agent.py, it shows ImportError: cannot import name 'ImpalaCNN' from 'tactile_gym.sb3_helpers.custom.custom_torch_layers'

ac-93 commented 2 years ago

Sorry this took so long, should be fixed in an updated version now.