search

Deci-AI / super-gradients

Easily train or fine-tune SOTA computer vision models with one open source training library. The home of Yolo-NAS.
https://www.supergradients.com
Apache License 2.0
4.54k stars 496 forks source link

How to custom YOLO-NAS model #1617

Closed YoungHsu1002 closed 8 months ago

YoungHsu1002 commented 11 months ago

💡 Your Question

Could you please advise on how I might modify the architecture of the model? I'm looking to make alterations and would appreciate some guidance on possible ways to approach this.

Like Two-stream-fusion detection.

Versions

No response

BloodAxe commented 11 months ago

What are the exact modifications to the model you would like to achieve? Two inputs & shared backbone or what?

CodingMonkey3688 commented 10 months ago

here: https://github.com/Deci-AI/super-gradients/blob/master/src/super_gradients/recipes/arch_params/yolo_nas_s_arch_params.yaml