How to Track ML Model Training: Tensorflow/Keras + Neptune Integration

1 min
Prince Canuma
5th April, 2022

What will you get with this integration?

TensorFlow is an open-source deep learning framework commonly used for building neural network models. Keras is an official higher-level API on top of TensorFlow. Neptune helps with keeping track of model training metadata.

With Neptune + TensorFlow / Keras integration you can:

  • Log hyperparameters for every run
  • See learning curves for losses and metrics during training
  • See hardware consumption and stdout/stderr output during training
  • Log TensorFlow tensors as images to see model predictions live
  • Log training code and git commit information
  • Log model weights

Check the documentation

Want to make your project easy to share and collaborate on?

    Contact with us

    This site is protected by reCAPTCHA and the Google Privacy Policy and Terms of Service apply.
    * - required fields