YOLOE: Fix Linear Probing example (#19985)

Signed-off-by: Mohammed Yasin <32206511+Y-T-G@users.noreply.github.com>
This commit is contained in:
Mohammed Yasin 2025-04-03 15:21:51 +06:00 committed by GitHub
parent f8bf96af2e
commit c2fbfc5c4a
No known key found for this signature in database
GPG key ID: B5690EEEBB952194

View file

@ -115,7 +115,6 @@ The YOLOE models are easy to integrate into your Python applications. Ultralytic
f"{head_index}.cv3.2.1",
]
)
freeze = [str(f) for f in range(0, len(model.model.model))]
model.train(
data="coco128-seg.yaml",