kunlunxin pytorch resnet50 add requirements.txt and environment_varia…

…bles.sh (#298)
FlagOpen · Oct 31, 2023 · 6fd0bed · 6fd0bed
1 parent ccb7bba
commit 6fd0bed
Show file tree

Hide file tree

Showing 3 changed files with 9 additions and 0 deletions.
diff --git a/training/kunlunxin/resnet50-pytorch/config/environment_variables.sh b/training/kunlunxin/resnet50-pytorch/config/environment_variables.sh
@@ -0,0 +1,4 @@
+export BKCL_PCIE_RING=1
+export BKCL_TIMEOUT=1800
+export XMLIR_D_XPU_L3_SIZE=66060288
+export XDNN_CONV_GEMM_DTYPE=int16
diff --git a/training/kunlunxin/resnet50-pytorch/config/requirements.txt b/training/kunlunxin/resnet50-pytorch/config/requirements.txt
@@ -0,0 +1 @@
+https://download.pytorch.org/whl/cpu/torchvision-0.13.1%2Bcpu-cp38-cp38-linux_x86_64.whl
diff --git a/training/run_benchmarks/config/test_conf.py b/training/run_benchmarks/config/test_conf.py
@@ -53,6 +53,7 @@
     "model:framework:hardwareID:nnodes:nproc:repeat": "dataset path"}
 '''
 CASES = {
+    # nvidia cases
     "bert:pytorch_1.8:A100:1:8:1": "/raid/home_datasets_ckpt/bert/train/",
     "glm:pytorch_1.8:A100:1:8:1": "/raid/home_datasets_ckpt/glm/train/",
     "cpm:pytorch_1.8:A100:1:8:1": "/raid/home_datasets_ckpt/cpm/train/",
@@ -98,4 +99,7 @@
     # "gpt3_13B:paddle_2.5.1:TP2PP1SH1SP4A10040G:1:8:1":"/raid/dataset/gpt-3/"
     # "gpt3_13B:paddle_2.5.1:TP2PP1SH2SP4A10040G:1:8:1":"/raid/dataset/gpt-3/"
     # "gpt3_13B:paddle_2.5.1:TP2PP4SH1SP1A10040G:1:8:1":"/raid/dataset/gpt-3/"
+
+    # kunlunxin cases
+    # "resnet50:pytorch:R300:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
 }
Original file line number	Diff line number	Diff line change
		@@ -0,0 +1 @@
		https://download.pytorch.org/whl/cpu/torchvision-0.13.1%2Bcpu-cp38-cp38-linux_x86_64.whl