Training the agent using the torch-rl repository

[MitulMagu]

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!

[bborj]

Were you able to fix the problem?