add PYTHONPATH

open-mmlab · Mar 7, 2022 · ef23c62 · ef23c62
1 parent 4623b10
commit ef23c62
Show file tree

Hide file tree

Showing 8 changed files with 8 additions and 0 deletions.
diff --git a/tools/mmcls/dist_search.sh b/tools/mmcls/dist_search.sh
@@ -5,5 +5,6 @@ CHECKPOINT=$2
 GPUS=$3
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/search_mmcls.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
diff --git a/tools/mmcls/dist_test.sh b/tools/mmcls/dist_test.sh
@@ -5,5 +5,6 @@ CHECKPOINT=$2
 GPUS=$3
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/test_mmcls.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
diff --git a/tools/mmcls/dist_train.sh b/tools/mmcls/dist_train.sh
@@ -4,5 +4,6 @@ CONFIG=$1
 GPUS=$2
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/train_mmcls.py $CONFIG --launcher pytorch ${@:3}
diff --git a/tools/mmdet/dist_search.sh b/tools/mmdet/dist_search.sh
@@ -5,5 +5,6 @@ CHECKPOINT=$2
 GPUS=$3
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/search_mmdet.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
diff --git a/tools/mmdet/dist_test.sh b/tools/mmdet/dist_test.sh
@@ -5,5 +5,6 @@ CHECKPOINT=$2
 GPUS=$3
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/test_mmdet.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
diff --git a/tools/mmdet/dist_train.sh b/tools/mmdet/dist_train.sh
@@ -4,5 +4,6 @@ CONFIG=$1
 GPUS=$2
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/train_mmdet.py $CONFIG --launcher pytorch ${@:3}
diff --git a/tools/mmseg/dist_test.sh b/tools/mmseg/dist_test.sh
@@ -5,5 +5,6 @@ CHECKPOINT=$2
 GPUS=$3
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/test_mmseg.py $CONFIG $CHECKPOINT --launcher pytorch ${@:4}
diff --git a/tools/mmseg/dist_train.sh b/tools/mmseg/dist_train.sh
@@ -4,5 +4,6 @@ CONFIG=$1
 GPUS=$2
 PORT=${PORT:-29500}
 
+PYTHONPATH="$(dirname $0)/../..":$PYTHONPATH \
 python -m torch.distributed.launch --nproc_per_node=$GPUS --master_port=$PORT \
     $(dirname "$0")/train_mmseg.py $CONFIG --launcher pytorch ${@:3}