Skip to content

Extract the YOLOv12 state_dict before upload #370

New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

Sign up for GitHub

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

Jump to bottom
Merged
lrosemberg merged 14 commits into from
Apr 3, 2025
Merged

Extract the YOLOv12 state_dict before upload #370

lrosemberg merged 14 commits into from
Apr 3, 2025

Conversation

@lrosemberg
Copy link
Contributor

@lrosemberg lrosemberg commented Apr 2, 2025 •
edited
Loading

Description

  • Extract the YOLOv12 state_dict before upload.
  • Also considers model["ema"], if model["model"] is none.

Type of change

  • Bug fix (non-breaking change which fixes an issue)

How has this change been tested, please provide a testcase or example of how you tested the change?

  • Locally and uploading do staging/prod

Any specific deployment considerations

  • Release package

@lrosemberg lrosemberg self-assigned this Apr 2, 2025
@lrosemberg lrosemberg requested a review from probicheaux April 3, 2025 01:21
tonylampada
tonylampada approved these changes Apr 3, 2025
View reviewed changes
Copy link
Collaborator

@tonylampada tonylampada left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

lgtm

@lrosemberg lrosemberg merged commit 8e3c9a1 into main Apr 3, 2025
8 checks passed
@iurisilvio iurisilvio deleted the lean/fix-yolov12-upload branch August 27, 2025 12:40
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Reviewers

@tonylampada tonylampada tonylampada approved these changes

@probicheaux probicheaux Awaiting requested review from probicheaux

Assignees

@lrosemberg lrosemberg

Labels

None yet

Projects

None yet

Milestone

No milestone

Development

Successfully merging this pull request may close these issues.

3 participants

@lrosemberg @tonylampada