-
Notifications
You must be signed in to change notification settings - Fork 517
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.
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
Feature/sg 849 add replace in channels #1557
Conversation
…nput_channels + adding docstings
…G-849-add_replace_in_channels
Most important thing this PR is missin, is a mechanism to replace the stem out of the box when using pretrained weights. |
Added here 287b28f |
Added update of |
src/super_gradients/training/models/detection_models/yolo_nas/yolo_stages.py
Outdated
Show resolved
Hide resolved
src/super_gradients/training/models/detection_models/yolo_nas/yolo_stages.py
Outdated
Show resolved
Hide resolved
44d9e93
to
df11873
Compare
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
LGTM
Adding
replace_input_channels
andget_input_channels
toSgModel
(and other required stems/backbones)