Neural ODE minibatch error with multi-dimension input condition

Specific Domains Machine Learning
,
6

It’s basically that calling the splat + adjoint creates a tuple of gradients, which are not the same type as the primal.

1 Like