world-modelz / dreamax

A scalable Dreamer implementation in JAX
MIT License
11 stars 2 forks source link

Fix wrong color channel order of videos/env_render video #17

Open andreaskoepf opened 2 years ago

andreaskoepf commented 2 years ago

The order of the color channels of the videos/env_render video seems to be wrong, in contrast to the other videos logged to tensorboard the environment is shown with wrong colors, see code https://github.com/world-modelz/dreamax/blob/1ff8933662a414dce6e42d2d1267e98b0f0f92ec/train.py#L203-L207