MammaM14
Results
1
comments of
MammaM14
Sorry, I forgot to mention that in the "learned_dynamics" function I had to remove "jnp.squeeze" jnp.squeeze(nn_forward_fn(params, state), axis=-1) because otherwise the code would not work. Sorry for the trouble, your...