Important: This video was created in December 2021. For the most up-to-date code examples, please refer to the Neptune-TensorFlow / Keras integration docs.
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
neptune.ai is an MLOps stack component for experiment tracking. So we’re constantly working on making it easy to integrate with other parts of the workflow.
It is already integrated with 25+ tools and libraries, and the list is growing. You can check our roadmap to see what’s currently under development.
Other useful resources
See also Keras integration guide.
More about How to Track ML Model Training: TensorFlow/Keras + neptune.ai Integration
What is a Project in Neptune?
Model Training: Detectron2 + neptune.ai Integration [Example]
Model Training: Prophet + neptune.ai Integration [Example]
Create AzureML Pipeline – Workshop with Aurimas Griciūnas
Explore more resources:
Content type
Area of interest
Area of interest
Couldn’t find the use case you were looking for?
Just get in touch, and our ML team will create a custom demo for you.