add faster_rcnn for kunlunxin (#167)

FlagOpen · Jul 26, 2023 · a2f083c · a2f083c
1 parent edf7ce2
commit a2f083c
Show file tree

Hide file tree

Showing 4 changed files with 19 additions and 0 deletions.
diff --git a/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x1.py b/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x1.py
@@ -0,0 +1,4 @@
+vendor: str = "kunlunxin"
+train_batch_size = 8
+eval_batch_size = 8
+lr = 0.16
diff --git a/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x8.py b/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x8.py
@@ -0,0 +1,4 @@
+vendor: str = "kunlunxin"
+train_batch_size = 8
+eval_batch_size = 8
+lr = 0.16
diff --git a/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x2x8.py b/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x2x8.py
@@ -0,0 +1,4 @@
+vendor: str = "kunlunxin"
+train_batch_size = 8
+eval_batch_size = 8
+lr = 0.16
diff --git a/training/kunlunxin/faster_rcnn-pytorch/config/environment_variables.sh b/training/kunlunxin/faster_rcnn-pytorch/config/environment_variables.sh
@@ -0,0 +1,7 @@
+# =================================================
+# Export variables
+# =================================================
+
+
+export XACC_ENABLE=1
+export BKCL_PCIE_RING=1