search

jurgisp / pydreamer

PyTorch implementation of DreamerV2 model-based RL algorithm
MIT License
205 stars 47 forks source link

Trouble with image_categorical, map_categorical for MiniGrid envs #5

Open stephmilani opened 2 years ago

stephmilani commented 2 years ago

Hi! I'm trying to use your codebase for various MiniGrid environments. Do you have any tips for setting the values for image_channels and map_channels?

If I just use the default config, I get an out-of-bounds error in the img_to_onehot function.

jiachengc commented 1 year ago

Hi! I'm trying to use your codebase for various MiniGrid environments. Do you have any tips for setting the values for image_channels and map_channels?

If I just use the default config, I get an out-of-bounds error in the img_to_onehot function.

Hi Steph, I got the same problem when I try to run on Mini-Grid environment about the map_channels. Have you fixed it?

jurgisp commented 1 year ago

@jiachengc could you provide more info on how you are installing the MiniGrid environment (which version?) and how you are launching the training (command line arguments), so I can try to reproduce? Thanks.