mlx-examples icon indicating copy to clipboard operation
mlx-examples copied to clipboard

Loader of pretrain Resnet on imageNet

Open thegodone opened this issue 1 year ago • 2 comments

Would be nice to have a loader of pretrain weights for Resnet on imageNet (1000 tasks) from keras or torch: https://pytorch.org/vision/main/models/resnet.html or https://www.tensorflow.org/api_docs/python/tf/keras/applications/resnet50/ResNet50

thegodone avatar Dec 18 '23 05:12 thegodone

@thegodone found some code for alexnet and vgg with loading of pretrained here. might provide hints on a loader https://github.com/robertmccraith/mimm/tree/main/mimm/models

bigsnarfdude avatar Dec 18 '23 17:12 bigsnarfdude

This would be nice to have as an example if anyone wants to contribute it. Once we have Batch Normalization in mlx.nn which is a PR right now, it should be pretty straightforward.

awni avatar Dec 20 '23 18:12 awni