NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
nnU-Net for PyTorch
Resource
NVIDIA Deep Learning Examples
NVIDIA Deep Learning Examples
nnU-Net for PyTorch

An optimized, robust and self-adapting framework for U-Net based medical image segmentation

Changelog

November 2022

  • Container updated to 22.11
  • Add support for 3D channel last convolutions
  • Add support for nvFuser Instance Normalization
  • Add support for GPU binding

October 2022

  • Add Jupyter Notebook with BraTS'22 solution (ranked 2)

December 2021

  • Container updated to 22.11
  • Use MONAI DynUNet instead of custom U-Net implementation
  • Add balanced multi-GPU evaluation
  • Support for evaluation with resampled volumes to original shape

October 2021

  • Add Jupyter Notebook with BraTS'21 solution (ranked 3)

May 2021

  • Add Triton Inference Server support
  • Removed deep supervision, attention, and drop block

March 2021

  • Container updated to 21.02
  • Change data format from tfrecord to npy and data loading for 2D

January 2021

  • Initial release
  • Add notebook with custom dataset loading

Known issues

There are no known issues in this release.