6 lines
340 B
Bash
6 lines
340 B
Bash
export CUDA_VISIBLE_DEVICES=4,5,6,7
|
|
python /data6/zhangw/yolov5/train.py \
|
|
--resume /data6/zhangw/ZW_gra_project/yolov5_result_pretrained/G10_decoupled_head_light_FGEM_safe_e1/weights/last.pt \
|
|
--epochs 2 \
|
|
--project /data6/zhangw/ZW_gra_project/yolov5_result_pretrained \
|
|
--name G10_decoupled_head_light_FGEM_safe_e2_resume |