Hi all,

I am trying to implement dreamer v3 on my custom environment with

action_space = gym.spaces.Box(low=-np.inf, high=np.inf, shape=(64,), dtype=np.float32)

observation_space = gym.spaces.Box(low=-np.inf, high=np.inf, shape=(204,), dtype=np.float32)

The following errors were raised

AttributeError: Exception encountered when calling layer ‘dreamer_model’

AttributeError: Exception encountered when calling layer ‘dynamics_predictor’

AttributeError: Exception encountered when calling layer ‘z32x32’

Call arguments received by layer ‘dreamer_model’ (type DreamerModel):

• inputs=None

• observations=tf.Tensor(shape=(1, 64, 204), dtype=float32)

• actions=tf.Tensor(shape=(1, 64, 64), dtype=float32)

• is_first=tf.Tensor(shape=(1, 64), dtype=bool)

• start_is_terminated_BxT=tf.Tensor(shape=(64,), dtype=bool)

Not too sure whether this might be because of my differ obs and action space instead of the standard 64,64,3, I have tried to use the dreamer catalogclass but it is still not working, any ideas?

many thanks