Implementing Keras custom layer in Core ML model with enabling GPU acceleration
Code for GPU computation in Apple Core ML model converted from Keras
Published in
4 min readMar 8, 2021
Backgrounds
I use a Windows machine with nVidia’s powerful GPU for training my Keras model with a huge dataset. This means that I must convert the trained Keras model into Apple’s Core…