You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
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!
The text was updated successfully, but these errors were encountered:
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!
The text was updated successfully, but these errors were encountered: