Add docker workload names

mlcommons · Dec 8, 2023 · 3071a30 · 3071a30
1 parent a8ad75e
commit 3071a30
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/docker/scripts/startup.sh b/docker/scripts/startup.sh
@@ -112,7 +112,8 @@ done
 # Check if arguments are valid
 VALID_DATASETS=("criteo1tb" "imagenet"  "fastmri" "ogbg" "librispeech" \
                 "wmt" "mnist")
-VALID_WORKLOADS=("criteo1tb" "imagenet_resnet" "imagenet_vit" "fastmri" "ogbg" \
+VALID_WORKLOADS=("criteo1tb" "imagenet_resnet" "imagenet_resnet_silu", "imagenet_resnet_gelu", \
+                 "imagenet_resnet_large_bn_init" "imagenet_vit" "fastmri" "ogbg" \
                  "wmt" "librispeech_deepspeech" "librispeech_conformer" "mnist" \
                  "criteo1tb_resnet" "criteo1tb_layernorm" "criteo_embed_init" \
                  "conformer_layernorm" "conformer_attention_temperature" \