How to Track ML Model Training: PyTorch + Neptune Integration

1 min
Prince Canuma
11th April, 2022

What will you learn?

You will learn how to use Neptune + PyTorch to help you keep track of your model training metadata.

With Neptune + PyTorch you can:

  • Log model configuration
  • Log hyperparameters
  • Log loss & metrics
  • Log training code and git information
  • Log images and the predictions
  • Log artifacts (i.e. model weights, dataset version)
  • Log 2-D/3-D tensors as images or 1-D tensors as metrics

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