Training a Segmentation model

Hi there,I’m a Computer Science student and I have a MacBook Pro 2019 and I’m thinking in buying a new Mac either a Mac Studio or a MacBook Pro but I want to use it for ML. I’m now doing a segmentation model and I’m wondering if I could use Core Ml or the Apple Neural Engine in the new M3 chips to train it, I’m now using colab and tensorflow to create the model but it’s not doing the job, I’m falling short of Cuda memory. Thanks :)

Hello @gozale6,

It is certainly possible to train a segmentation model on a Mac, though image segmentation is not something that we have a specific Create ML template for.

It sounds like you are utilizing TensorFlow to train your model right now, so if you do decide to train on Mac, you may want to investigate the tensorflow-metal plugin: https://developer.apple.com/metal/tensorflow-plugin/

Best regards,

Greg

Training a Segmentation model
 
 
Q