From 58d085bb1b205e256b4fba89dd66f5cd529c29e8 Mon Sep 17 00:00:00 2001 From: Reiase Date: Tue, 25 Jul 2023 15:13:15 +0800 Subject: [PATCH] add faster_rcnn for kunlunxin --- .../faster_rcnn-pytorch/config/config_R200x1x1.py | 4 ++++ .../faster_rcnn-pytorch/config/config_R200x1x8.py | 4 ++++ .../faster_rcnn-pytorch/config/config_R200x2x8.py | 4 ++++ .../faster_rcnn-pytorch/config/environment_variables.sh | 7 +++++++ 4 files changed, 19 insertions(+) create mode 100644 training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x1.py create mode 100644 training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x8.py create mode 100644 training/kunlunxin/faster_rcnn-pytorch/config/config_R200x2x8.py create mode 100755 training/kunlunxin/faster_rcnn-pytorch/config/environment_variables.sh diff --git a/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x1.py b/training/kunlunxin/faster_rcnn-pytorch/config/config_R200x1x1.py new file mode 100644 index 000000000..23f715329 --- /dev/null +++ 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 new file mode 100644 index 000000000..23f715329 --- /dev/null +++ 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 new file mode 100644 index 000000000..23f715329 --- /dev/null +++ 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 new file mode 100755 index 000000000..b368e793f --- /dev/null +++ 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