@@ -15,21 +15,21 @@ These examples, along with our NVIDIA deep learning software stack, are provided
1515## Computer Vision
1616| Models | Framework | A100 | AMP | Multi-GPU | Multi-Node | TRT | ONNX | Triton | TF-TRT | NB |
1717| ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- | ------------- |
18- | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnet50v1.5 ) | PyTorch | Yes | Yes | Yes | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnet50 ) | - | - |
19- | [ ResNeXt101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnext101-32x4d ) | - | - |
20- | [ SEResNeXt101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/se-resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/se-resnext101-32x4d ) | - | - |
18+ | [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnet50v1.5 ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnet50 ) | - | - |
19+ | [ ResNeXt-101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/resnext101-32x4d ) | - | - |
20+ | [ SE-ResNeXt-101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/se-resnext101-32x4d ) | PyTorch | Yes | Yes | Yes | - | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Classification/ConvNets/triton/se-resnext101-32x4d ) | - | - |
2121| [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Segmentation/MaskRCNN ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Segmentation/MaskRCNN/pytorch/notebooks/pytorch_MaskRCNN_pyt_train_and_inference.ipynb ) |
2222| [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Detection/SSD ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/PyTorch/Detection/SSD/examples/inference.ipynb ) |
2323| [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnet50v1.5 ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
2424| [ ResNeXt101] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
25- | [ SEResNeXt101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/se-resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
25+ | [ SE-ResNeXt-101 ] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Classification/ConvNets/se-resnext101-32x4d ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
2626| [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
2727| [ SSD] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Detection/SSD ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/Detection/SSD/models/research/object_detection/object_detection_tutorial.ipynb ) |
28- | [ U-Net Ind] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial ) | TensorFlow | Yes | Yes | Yes | - | Yes | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial/notebooks ) |
29- | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Medical ) | TensorFlow | Yes | Yes | Yes | - | Yes | - | - | Yes | - |
30- | [ U-Net 3D] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_3D_Medical ) | TensorFlow | Yes | Yes | Yes | - | Yes | - | - | Yes | - |
31- | [ V-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/VNet ) | TensorFlow | Yes | Yes | Yes | - | Yes | - | - | Yes | - |
32- | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/UNet_Medical ) | TensorFlow2 | Yes | Yes | Yes | - | Yes | - | - | Yes | - |
28+ | [ U-Net Ind] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | Yes | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Industrial/notebooks ) |
29+ | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
30+ | [ U-Net 3D] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/UNet_3D_Medical ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
31+ | [ V-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/Segmentation/VNet ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
32+ | [ U-Net Med] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/UNet_Medical ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | - | - |
3333| [ Mask R-CNN] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/Segmentation/MaskRCNN ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | - | - |
3434| [ ResNet-50] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/MxNet/Classification/RN50v1.5 ) | MXNet | - | Yes | Yes | - | - | - | - | - | - |
3535
@@ -40,7 +40,7 @@ These examples, along with our NVIDIA deep learning software stack, are provided
4040| [ TransformerXL] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/LanguageModeling/Transformer-XL ) | PyTorch | Yes | Yes | Yes | Yes | - | - | - | - | - |
4141| [ GNMT] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Translation/GNMT ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | - |
4242| [ Transformer] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/PyTorch/Translation/Transformer ) | PyTorch | Yes | Yes | Yes | - | - | - | - | - | - |
43- | [ ELECTRA] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/LanguageModeling/ELECTRA ) | TensorFlow2 | Yes | Yes | Yes | - | - | - | - | - | - |
43+ | [ ELECTRA] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow2/LanguageModeling/ELECTRA ) | TensorFlow2 | Yes | Yes | Yes | Yes | - | - | - | - | - |
4444| [ BERT] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/LanguageModeling/BERT ) | TensorFlow | Yes | Yes | Yes | Yes | Yes | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/LanguageModeling/BERT/triton ) | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/LanguageModeling/BERT/notebooks ) |
4545| [ BioBert] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/LanguageModeling/BERT/biobert ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | [ Yes] ( https://github.com/NVIDIA/DeepLearningExamples/blob/master/TensorFlow/LanguageModeling/BERT/notebooks/biobert_ner_tf_inference.ipynb ) |
4646| [ TransformerXL] ( https://github.com/NVIDIA/DeepLearningExamples/tree/master/TensorFlow/LanguageModeling/Transformer-XL ) | TensorFlow | Yes | Yes | Yes | - | - | - | - | - | - |
0 commit comments