2021-12-19 10:34:06 +00:00
---
2021-12-27 08:41:43 +00:00
title: MegEngine
2021-12-19 10:34:06 +00:00
---
The [megengine_mnist.py ](https://github.com/allegroai/clearml/blob/master/examples/frameworks/megengine/megengine_mnist.py )
example demonstrates the integration of ClearML into code that uses [MegEngine ](https://github.com/MegEngine/MegEngine )
and [TensorBoardX ](https://github.com/lanpa/tensorboardX ). ClearML automatically captures models saved with `megengine` .
The example script does the following:
2024-12-09 13:02:44 +00:00
* Trains a simple deep neural network on MegEngine's built-in [MNIST ](https://www.megengine.org.cn/doc/master/en/reference/api/megengine.data.dataset.MNIST.html )
2021-12-19 10:34:06 +00:00
dataset.
* Creates a TensorBoardX `SummaryWriter` object to log scalars during training.
2025-02-06 15:31:11 +00:00
* Creates a ClearML task named `megengine mnist train` in the `examples` project.
2021-12-19 10:34:06 +00:00
## Hyperparameters
2025-02-06 15:31:11 +00:00
ClearML automatically logs command line options defined with `argparse` . They appear in the task's **CONFIGURATION**
2023-01-12 10:49:55 +00:00
tab under **HYPERPARAMETERS** ** >** **Args** .
2021-12-19 10:34:06 +00:00

## Scalars
The example script's `train` function calls TensorBoardX's `SummaryWriter.add_scalar` method to report `loss` .
ClearML automatically captures the data that is added to the `SummaryWriter` object.
These scalars can be visualized in plots, which appear in the ClearML [WebApp ](../../../webapp/webapp_home.md ), in the
2025-02-06 15:31:11 +00:00
task's **SCALARS** tab.
2021-12-19 10:34:06 +00:00

## Models
ClearML automatically captures the model logged using the `megengine.save` method, and saves it as an artifact.
2025-02-06 15:31:11 +00:00
View saved snapshots in the task's **ARTIFACTS** tab.
2021-12-19 10:34:06 +00:00

To view the model details, click the model name in the **ARTIFACTS** page, which will open the model's info tab. Alternatively, download the model.
The model info panel contains the model details, including:
* Model URL
* Framework
* Snapshot locations.

## Console
2025-02-06 15:31:11 +00:00
All console output during the script's execution appears in the task's **CONSOLE** page.
2024-08-05 07:12:18 +00:00
2021-12-19 10:34:06 +00:00
