update kunlunxin transformer_xl config (#307)

FlagOpen · Nov 3, 2023 · f4bd689 · f4bd689
1 parent d199125
commit f4bd689
Show file tree

Hide file tree

Showing 3 changed files with 4 additions and 1 deletion.
diff --git a/training/kunlunxin/transformer_xl-pytorch/config/environment_variables.sh b/training/kunlunxin/transformer_xl-pytorch/config/environment_variables.sh
@@ -1,2 +1,4 @@
-export XACC_ENABLE=1
+export XACC=1
+
 export BKCL_PCIE_RING=1
+export BKCL_TIMEOUT=1800
diff --git a/training/kunlunxin/transformer_xl-pytorch/extern/.gitkeep b/training/kunlunxin/transformer_xl-pytorch/extern/.gitkeep
diff --git a/training/run_benchmarks/config/test_conf.py b/training/run_benchmarks/config/test_conf.py
@@ -105,5 +105,6 @@
     # kunlunxin cases
     # "gpt2:pytorch:R300:1:8:1": "/raid/dataset/gpt2"
     # "resnet50:pytorch:R300:1:8:1": "/raid/dataset/ImageNet_1k_2012/",
+    # "transformer_xl:pytorch:R300:1:8:1": "/raid/dataset/transformer_xl/",
 }