Skip to content

Commit 17bbdaf

Browse files
authored
1017 add fast_training tutorial (#22)
* [DLMED] add fast_training tutorial Signed-off-by: Nic Ma <[email protected]> * [DLMED] update README * [DLMED] complete fast training tutorial Signed-off-by: Nic Ma <[email protected]> * [DLMED] update according to comments Signed-off-by: Nic Ma <[email protected]>
1 parent f94c940 commit 17bbdaf

File tree

2 files changed

+680
-0
lines changed

2 files changed

+680
-0
lines changed

README.md

Lines changed: 5 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -28,6 +28,9 @@ And compares the training speed and memory usage with/without AMP.
2828
This tutorial shows how to construct a training workflow of multi-labels segmentation task based on [MSD Brain Tumor dataset](http://medicaldecathlon.com).
2929
#### [dataset_type_performance](./dataset_type_performance.ipynb)
3030
This notebook compares the performance of `Dataset`, `CacheDataset` and `PersistentDataset`. These classes differ in how data is stored (in memory or on disk), and at which moment transforms are applied.
31+
#### [fast_training_tutorial](./fast_training_tutorial.ipynb)
32+
This tutorial compares the training performance of pure PyTorch program and optimized program in MONAI based on NVIDIA GPU device and latest CUDA library.
33+
The optimization methods mainly include: `AMP`, `CacheDataset` and `Novograd`.
3134
#### [integrate_3rd_party_transforms](./integrate_3rd_party_transforms.ipynb)
3235
This tutorial shows how to integrate 3rd party transforms into MONAI program.
3336
Mainly shows transforms from BatchGenerator, TorchIO, Rising and ITK.
@@ -48,6 +51,8 @@ This tutorial shows how to leverage `EnsembleEvaluator`, `MeanEnsemble` and `Vot
4851
This notebook is a quick demo for devices, run the Ignite trainer engine on CPU, GPU and multiple GPUs.
4952
#### [nifti_read_example](./nifti_read_example.ipynb)
5053
Illustrate reading NIfTI files and iterating over image patches of the volumes loaded from them.
54+
#### [nnunet_tutorial](./nnunet_tutorial.ipynb)
55+
This tutorial shows how to train 3D segmentation tasks on all the 10 decathlon datasets with the reimplementation of NNUnet in MONAI.
5156
#### [post_transforms](./post_transforms.ipynb)
5257
This notebook shows the usage of several post transforms based on the model output of spleen segmentation task.
5358
#### [public_datasets](./public_datasets.ipynb)

0 commit comments

Comments
 (0)