Skip to content
Snippets Groups Projects
Unverified Commit 13c7f5ae authored by Xuan Gu's avatar Xuan Gu Committed by GitHub
Browse files

Update README.md

parent 8e25315a
No related branches found
No related tags found
No related merge requests found
......@@ -54,3 +54,13 @@ bash benchmark_nnunet_pytorch_berzelius.sh
cd /proj/nsc/xuan/ngc/DeepLearningExamples/PyTorch/Segmentation/nnUNet
sbash benchmark_nnunet_pytorch_berzelius_multi_node.sh
```
#### Results
We collect benchmark results of throughput (images/sec) for
- Dimention = 2
- Nodes = 1, 2
- GPUs = 1 - 8 (for 1 node), 16 (for 2 nodes)
- Batch size = 2, 4, 8, 16, 32, 64, 128
We run 100 iterations for each set of parameters.
- Observation 1: throughput_tf32 > throughput_amp when batch_size is small (1, 2, 4, 8); throughput_tf32 < throughput_amp when batch_size is large (16, 32, 64, 128).
- Observation 2: The coefficient of variation for the 100 iteration is smallest when batch_size = 128.
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment