mlx icon indicating copy to clipboard operation
mlx copied to clipboard

Pretrained CV Models available?

Open remainedmind opened this issue 2 years ago • 1 comments

Is there any way to import pretrained SOTA CV models (e.g. MobileNet) instead of creating toy CV models? I didn't find examples. Thanks for any response

remainedmind avatar Dec 09 '23 07:12 remainedmind

I have implemented VGG with loading from Pytorch weights here: https://github.com/robertmccraith/mimm there's some training code also, more models to come soon.

robertmccraith avatar Dec 12 '23 21:12 robertmccraith