From 518ea117ead8c5b4a2296d179cef2e185a03508b Mon Sep 17 00:00:00 2001 From: Kathrine Date: Fri, 1 Mar 2024 11:36:35 +0800 Subject: [PATCH] Update test_conf.py [metax] add bert_hf /glm sample --- training/run_benchmarks/config/test_conf.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/training/run_benchmarks/config/test_conf.py b/training/run_benchmarks/config/test_conf.py index 9a7eb1c78..94d9f81f1 100644 --- a/training/run_benchmarks/config/test_conf.py +++ b/training/run_benchmarks/config/test_conf.py @@ -153,3 +153,6 @@ # "resnet50:pytorch_2.0:C500:1:8:1": "/raid/dataset/ImageNet_1k_2012/", # "swin_transformer:pytorch_2.0:C500:1:8:1": "/raid/dataset/ImageNet_1k_2012/", # "WaveGlow:pytorch_2.0:C500:1:8:1": "/raid/dataset/LJSpeech/", + # "bert_hf:pytorch_2.0:C500:1:8:1": "/raid/dataset/bert_hf_train", + # "glm:pytorch_2.0:C500:1:8:1": "/raid/home_datasets_ckpt/glm/train/", +