I am trying to train the agents in the multigrid environment using the torch-rl repository (https://github.com/lcswillems/rl-starter-files) which is compatible with gym-minigrid. After registering the new environment in gym (multigrid-collect-v0) and running the environment using
I am receiving the following runtime error: RuntimeError: Expected 4-dimensional input for 4-dimensional weight [16, 3, 2, 2], but got 5-dimensional input of size [16, 3, 5, 3, 6] instead
Is there a way to make it compatible with the torch-rl repository or am I making a mistake somewhere? Any help would be much appreciated, thanks!
I am trying to train the agents in the multigrid environment using the torch-rl repository (https://github.com/lcswillems/rl-starter-files) which is compatible with gym-minigrid. After registering the new environment in gym (multigrid-collect-v0) and running the environment using
python3 -m scripts.train --algo ppo --env multigrid-collect-v0
I am receiving the following runtime error: RuntimeError: Expected 4-dimensional input for 4-dimensional weight [16, 3, 2, 2], but got 5-dimensional input of size [16, 3, 5, 3, 6] instead
Is there a way to make it compatible with the torch-rl repository or am I making a mistake somewhere? Any help would be much appreciated, thanks!