Skip to main content

TensorFlow MNIST

The tensorflow_mnist.py example demonstrates the integration of ClearML into code that uses TensorFlow and Keras to train a neural network on the Keras built-in MNIST handwritten digit dataset.

When the script runs, it creates a task named Tensorflow v2 mnist with summaries in the examples project.

Scalars​

The loss and accuracy metric scalar plots appear in the task's page in the ClearML web UI under SCALARS. Resource utilization plots, which are titled :monitor: machine, also appear in the SCALARS tab.

Task scalars Task scalars

Hyperparameters​

ClearML automatically logs TensorFlow Definitions. They appear in CONFIGURATION > HYPERPARAMETERS > TF_DEFINE.

Task hyperparameters Task hyperparameters

Console​

All console output appears in CONSOLE.

Task console Task console

Artifacts​

Models created by the task appear in the task's ARTIFACTS tab. ClearML automatically logs and tracks models and any snapshots created using TensorFlow.

Task models Task models

Clicking on a model's name takes you to the model's page, where you can view the model's details and access the model.

Model details Model details