Skip to content

Release v1.0.0

Latest
Compare
Choose a tag to compare
@github-actions github-actions released this 28 Mar 13:32

First public release - v1.0.0!

Summary

tensorflow-hooks (importable as tf_hooks) provides PyTorch-like hooks for TensorFlow Keras layers. It provides two main functions:

  • register_forward_hook for forward hooks, which execute after forward inference of the layer.
  • register_forward_pre_hook for forward pre-hooks, which execute prior to forward inference of the layer.

See the README for more details.