How to track ML Model Training: Colab + Neptune Integration

1 min
Siddhant Sadangi
11th April, 2022

What will you get with this integration?

Google Colab is a temporary runtime environment. This means you lose all your data (unless saved externally) once you restart your kernel.

This is where you can leverage Neptune. By running model training on Google Colab and keeping track of it with Neptune you can log and download things like:

  • Parameters
  • Metrics and losses
  • Images, interactive charts, and other media
  • Hardware consumption
  • Model checkpoints and other artifacts

By doing that you can keep your run metadata safe even when the Google Colab kernel has died.

Read 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