"add DGXH100 platform option"

NVIDIA · Jan 16, 2025 · ac9133c · ac9133c
1 parent 729963d
commit ac9133c
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/PyTorch/Classification/ConvNets/configs.yml b/PyTorch/Classification/ConvNets/configs.yml
@@ -30,7 +30,10 @@ platform:
         workers: 10
         prefetch: 4
         gpu_affinity: socket_unique_contiguous
-
+    DGXH100:
+        workers: 10
+        prefetch: 4
+        gpu_affinity: socket_unique_contiguous
 mode:
     benchmark_training: &benchmark_training
         print_freq: 1