search

danijar / dreamerv3

Mastering Diverse Domains through World Models
https://danijar.com/dreamerv3
MIT License
1.36k stars 231 forks source link

mlp_keys, cnn_keys on customize envs #27

Closed janetwise closed 6 months ago

janetwise commented 1 year ago

What mlp_keys and cnn__keys should be used on low dimensional env input such as cartpole? thanks a lot!

danijar commented 1 year ago

Hey, that depends on the obs_key that's specified in the FromGym wrapper (which is used when turning the observation space of envs that only have one observation into a dictionary space).