Skip to content

Commit

Permalink
add iluvatar tips in test_conf.py
Browse files Browse the repository at this point in the history
  • Loading branch information
杨zhichao.yang committed Jan 19, 2024
1 parent d21fd57 commit 385866f
Showing 1 changed file with 2 additions and 4 deletions.
6 changes: 2 additions & 4 deletions training/run_benchmarks/config/test_conf.py
Original file line number Diff line number Diff line change
Expand Up @@ -136,16 +136,14 @@
# "bigtransfer:pytorch:BI-V100:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
# "vit:pytorch:BI-V100:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
# "transformer:pytorch:BI-V100:1:8:1": "/raid/dataset/transformer/wmt14_en_de_joined_dict",

# "bert_hf:pytorch:BI-V100:1:8:1": "/raid/dataset/bert_hf_train",
# "t5_small:pytorch:BI-V100:1:8:1": "/raid/dataset/t5_small",

# mthreads cases
# "resnet50:pytorch_2.0:S4000:1:8:1": "/data/flagperf/ImageNet",
# "retinanet:pytorch_2.0:S4000:1:8:1": "/data/flagperf/coco2017",
# "bert_hf:pytorch_2.0:S4000:1:8:1": "/data/flagperf/bert_hf",
# "llama2_7b:deepspeed:S4000:1:8:1": "/data/flagperf/llama/openwebtext",

# iluvatar cases
# "transformer:pytorch:BI-V100:1:8:1": "/raid/dataset/transformer/wmt14_en_de_joined_dict",
# "bert_hf:pytorch:BI-V100:1:8:1": "/raid/dataset/bert_hf_train",

}

0 comments on commit 385866f

Please sign in to comment.