Update test_conf.py

[Metax] add distilbert and cpm
FlagOpen · Mar 20, 2024 · 737c297 · 737c297
1 parent 3d765ff
commit 737c297
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/training/run_benchmarks/config/test_conf.py b/training/run_benchmarks/config/test_conf.py
@@ -163,6 +163,10 @@
     # "mobilenetv2:pytorch_2.0:C500:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
     # "mask_rcnn:pytorch_2.0:C500:1:8:1": "/raid/dataset/coco2017/",
     # "detr:pytorch_2.0:C500:1:8:1": "/raid/dataset/coco2017/",
+    # "cpm:pytorch_2.0:A100:1:8:1": "/raid/dataset/cpm/train/",
+    # "distilbert:pytorch_2.0:A100:1:8:1": "/raid/dataset/distilbert/",
+
+
 
     # dcu cases
     # "glm:pytorch_1.13:K100:1:8:1": "/home/chenych/datasets/glm_train_datset/",