Gradient update contexts are missing arrays except for the loss

ahrefs / ocannl

OCANNL: OCaml Compiles Algorithms for Neural Networks Learning

BSD 2-Clause "Simplified" License

67 stars 2 forks source link

Closed lukstafi closed 7 months ago

lukstafi commented 8 months ago

In particular, all gradients should be in the contexts as local arrays.

lukstafi commented 7 months ago

This got fixed recently.