fix(configs): update training epochs in film.yaml and film.yaml for classifier and YOLO models
This commit is contained in:
@@ -54,7 +54,7 @@ model:
|
|||||||
frozen_backbone: false # true = 冻结骨架只训分类头(线性探针)
|
frozen_backbone: false # true = 冻结骨架只训分类头(线性探针)
|
||||||
|
|
||||||
train:
|
train:
|
||||||
epochs: 30
|
epochs: 60
|
||||||
amp: auto
|
amp: auto
|
||||||
clip_grad: 5.0
|
clip_grad: 5.0
|
||||||
log_interval: 20
|
log_interval: 20
|
||||||
|
|||||||
@@ -23,7 +23,7 @@ model:
|
|||||||
weights: yolo26n.pt
|
weights: yolo26n.pt
|
||||||
|
|
||||||
train:
|
train:
|
||||||
epochs: 100
|
epochs: 500
|
||||||
imgsz: 640
|
imgsz: 640
|
||||||
batch: 16
|
batch: 16
|
||||||
workers: 8
|
workers: 8
|
||||||
|
|||||||
Reference in New Issue
Block a user