MOD/train.sh

9 lines
336 B
Bash

python train.py \
--data data/multispectral/uav.yaml\
--batch-size 32 \
--img-size 640 \
--device 0,1 \
--epochs 30 \
--weights train/exp/weights/best.pt \
--cfg models/transformer/yolov5l_fusion_transformerx3_llvip.yaml \