diff --git a/configs/classify/film.yaml b/configs/classify/film.yaml index c27abde..29e5d67 100644 --- a/configs/classify/film.yaml +++ b/configs/classify/film.yaml @@ -54,7 +54,7 @@ model: frozen_backbone: false # true = 冻结骨架只训分类头(线性探针) train: - epochs: 30 + epochs: 60 amp: auto clip_grad: 5.0 log_interval: 20 diff --git a/configs/detect/film.yaml b/configs/detect/film.yaml index cd06931..968136b 100644 --- a/configs/detect/film.yaml +++ b/configs/detect/film.yaml @@ -23,7 +23,7 @@ model: weights: yolo26n.pt train: - epochs: 100 + epochs: 500 imgsz: 640 batch: 16 workers: 8