mlx-examples
mlx-examples copied to clipboard
Loader of pretrain Resnet on imageNet
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 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
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.