edit pytorch lightning example

This commit is contained in:
Revital
2021-08-19 14:40:15 +03:00
parent c19af210de
commit 2e66ce4ad9
3 changed files with 7 additions and 3 deletions

View File

@@ -7,18 +7,21 @@ script demonstrates the integration of ClearML into code that uses PyTorch Light
The example script does the following:
* Trains a simple deep neural network on the PyTorch built-in MNIST dataset
* Specifies Argparse configurations which are automatically captured by ClearML
* Defines Argparse command line options, which are automatically captured by ClearML
* Creates an experiment named `pytorch lightning mnist example`, which is associated with the `examples` project.
## Scalars
The test loss and validation loss metric scalar plots appear in the experiment's page in the ClearML web UI under **RESULTS > SCALARS**.
Resource utilization plots, which are titled **:monitor: machine**, also appear in the **SCALARS** tab.
![PyTorch Lightning console](../../../img/examples_pytorch_lightning_scalars.png)
## Hyperparameters
Lightning returns logs after
ClearML automatically logs command line options defined with argparse. They appear in CONFIGURATIONS > HYPER PARAMETERS > Args.
ClearML automatically logs command line options defined with argparse and TensorFlow Definitions, which appear in
**CONFIGURATIONS > HYPER PARAMETERS > Args** and **TF_DEFINE** respectively.
![PyTorch Lightning parameters](../../../img/examples_pytorch_lightning_params.png)

Binary file not shown.

Before

Width:  |  Height:  |  Size: 83 KiB

After

Width:  |  Height:  |  Size: 83 KiB

View File

@@ -75,6 +75,7 @@ module.exports = {
'guides/frameworks/pytorch/tensorboard_toy_pytorch']
},
{'Pytorch Ignite': ['guides/frameworks/pytorch ignite/integration_pytorch_ignite']},
{'Pytorch Lightning': ['guides/frameworks/pytorch_lightning/pytorch_lightning_example']},
{
'Pytorch Notebooks': [
{'Audio': ['guides/frameworks/pytorch/notebooks/audio/audio_classification_UrbanSound8K', 'guides/frameworks/pytorch/notebooks/audio/audio_preprocessing_example']},