add ogbg variants to docker startup.sh

mlcommons · Jan 2, 2024 · fdcb5aa · fdcb5aa
1 parent d604850
commit fdcb5aa
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docker/scripts/startup.sh b/docker/scripts/startup.sh
@@ -119,7 +119,7 @@ VALID_WORKLOADS=("criteo1tb" "imagenet_resnet" "imagenet_resnet_silu" "imagenet_
                  "criteo1tb_resnet" "criteo1tb_layernorm" "criteo1tb_embed_init" \
                  "conformer_layernorm" "conformer_attention_temperature" \
                  "conformer_gelu" "fastmri_model_size" "fastmri_tanh" \
-                 "fastmri_layernorm")
+                 "fastmri_layernorm" "ogbg_gelu" "ogbg_silu" "ogbg_model_size")
 
 # Set data and experiment paths
 ROOT_DATA_BUCKET="gs://mlcommons-data"