cls_head_enhance_Lightweight

train_my_gra.sh

@@ -1,4 +1,4 @@
-export CUDA_VISIBLE_DEVICES=4,5,6,7
+export CUDA_VISIBLE_DEVICES=6,7
 python /data6/zhangw/yolov5/train.py \
     --img 960 \
     --batch 16 \
@@ -6,5 +6,5 @@ python /data6/zhangw/yolov5/train.py \
     --data /data6/zhangw/ZW_gra_project/CUB_100/cub100.yaml \
     --weights /data6/zhangw/yolov5/yolov5s6.pt \
     --project /data6/zhangw/ZW_gra_project/yolov5_result_pretrained \
-    --name G8_decoupled_head_light_e300 \
-    --cfg /data6/zhangw/yolov5/models/yolov5s6_decoupled_light.yaml
+    --name G11_cls_head_enhance_Lightweight_e300 \
+    --cfg /data6/zhangw/yolov5/models/yolov5s6_cls_enhance.yaml