What is the easiest way to run experiments on colored Atari environments?

[alversafa]

Hi,

For a while, I have been trying to run experiments on colored Atari environments with MuZero (that is when grayscale=False). However, I am experiencing a hard time on changing the configs in zoo/atari to do this. What would be the easiest way to do this?

In particular, I have set grayscale=False in the environment file. But I am wondering how to change the obs_shape and observation_shape parameters in the config file. And, I am also wondering whether if there is anything else that I would need to change?

Best

[alversafa]

I have solved the problem. It turns out that I also had to change the image_channel parameter in the policy file.