|
@@ -3,9 +3,9 @@ DATASET="coco"
|
|
|
DATA_ROOT="/data/datasets/"
|
|
DATA_ROOT="/data/datasets/"
|
|
|
|
|
|
|
|
# MODEL setting
|
|
# MODEL setting
|
|
|
-MODEL="rtcdet_s"
|
|
|
|
|
-IMAGE_SIZE=640
|
|
|
|
|
|
|
+MODEL=$1
|
|
|
RESUME="None"
|
|
RESUME="None"
|
|
|
|
|
+IMAGE_SIZE=640
|
|
|
if [[ $MODEL == *"yolov8"* ]]; then
|
|
if [[ $MODEL == *"yolov8"* ]]; then
|
|
|
# Epoch setting
|
|
# Epoch setting
|
|
|
BATCH_SIZE=128
|
|
BATCH_SIZE=128
|
|
@@ -72,8 +72,8 @@ else
|
|
|
fi
|
|
fi
|
|
|
|
|
|
|
|
# -------------------------- Train Pipeline --------------------------
|
|
# -------------------------- Train Pipeline --------------------------
|
|
|
-WORLD_SIZE=$1
|
|
|
|
|
-MASTER_PORT=$2
|
|
|
|
|
|
|
+WORLD_SIZE=$2
|
|
|
|
|
+MASTER_PORT=$3
|
|
|
if [ $WORLD_SIZE == 1 ]; then
|
|
if [ $WORLD_SIZE == 1 ]; then
|
|
|
python train.py \
|
|
python train.py \
|
|
|
--cuda \
|
|
--cuda \
|
yjh0410