maxmustermann/DeepLearningExamples
1
0
Fork 0
Code Issues Pull requests Projects Releases Wiki Activity

Merge pull request #893 from NVIDIA/pribalta/fixes_to_nnunet_readme

Browse source 
nnUNet: Add reference to PyTorch Lightning to readme
This commit is contained in:
nv-kkudrynski 2021-04-06 11:40:28 +02:00 committed by GitHub
parent a64351f3f1 86ae47f0db
commit a71bae7cda
No known key found for this signature in database
GPG key ID: 4AEE18F83AFDEB23
1 changed files with 2 additions and 0 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
PyTorch/Segmentation/nnUNet/README.md
View file

@ -53,6 +53,8 @@ The differences between this nnU-net and [original model](https://github.com/MIC
This model is trained with mixed precision using Tensor Cores on Volta, Turing, and the NVIDIA Ampere GPU architectures. Therefore, researchers can get results 2x faster than training without Tensor Cores, while experiencing the benefits of mixed precision training. This model is tested against each NGC monthly container release to ensure consistent accuracy and performance over time.
We developed the model using [PyTorch Lightning](https://ai.facebook.com/blog/reengineering-facebook-ais-deep-learning-platforms-for-interoperability/), a new easy to use framework that ensures code readability and reproducibility without the boilerplate.
### Model architecture
    
The nnU-Net allows training two types of networks: 2D U-Net and 3D U-Net to perform semantic segmentation of 3D images, with high accuracy and performance.