Skip to content

v1.0.8

Compare
Choose a tag to compare
@sokrypton sokrypton released this 09 Oct 23:25
· 263 commits to main since this release
c90c4d3
  • custom callback functions ([pre|loss|pos]_callback) have been refactored to be more flexible.
    • Supported input arguments include: ["inputs", "outputs", "params", "opt", "seq", "aux", "key"].
    • The pre_callback function can be used to modify inputs before prediction, loss_callback to add custom loss.
  • adding support for Optax optimizers

Full Changelog: v1.0.7...v1.0.8