how can I train a new student model rather than only resnet?

NVIDIA-AI-IOT / nanosam

A distilled Segment Anything (SAM) model capable of running real-time with NVIDIA TensorRT

Apache License 2.0

616 stars 52 forks source link

how can I train a new student model rather than only resnet? #19

Open countlesswu opened 8 months ago

countlesswu commented 8 months ago

so excellent work!!! but I need to train a student model like myself Lightweight vit model using the mobilesam teather model. how can I train a new student model rather than only resnet? can you give me some help? thank you!

countlesswu commented 7 months ago

ok fine.. i have solved it,thanks..

walywalid commented 3 months ago

how did you do so?