search

tracel-ai / models

Models and examples built with Burn
Apache License 2.0
180 stars 24 forks source link

MobileNetv2 #26

Closed Arjun31415 closed 6 months ago

Arjun31415 commented 6 months ago

Work in progress. Model structure seems to be correct, could be simplified more by getting rid of a few enums.

Need help in loading pytorch weight 🥲

laggui commented 6 months ago

@Arjun31415 I've made some changes to support the pre-trained weights and added an example which loads the weights for inference. Fixed some bugs that I uncovered when trying to load the weights, but otherwise mainly the form (missing README, NOTICES, added structs doc).

I didn't put the IMAGENET1K_V1 weights since the official weights at this url cannot be loaded via candle-core (zip-rs issue: "invalid Zip archive: Could not find central directory end").

@antimora @nathanielsimard since I contributed, do any of you mind reviewing this as well?

laggui commented 6 months ago

@Arjun31415 just merged 🎉 Thanks for your contribution!

Arjun31415 commented 6 months ago

time for mobilenetv3 :D Or fasterRCNN