Post

Replies

Boosts

Views

Activity

Reply to How to Train and Deploy PyTorch Models on Apple Hardware: A Unified Path for Deep ML Practice on Core ML?
I am doing some experiments in past few days, it's not easy to convert my PyTorch model to coreML models. I have to modify the python code to avoid conversation error, even if the conversation succeeds, it may crash when running in Xcode; finally I can manage to convert the model successfully and run it in Xcode, but it only uses the CPU to compute, only if I use fixed input shape, it can use the NPU to compute, but I really need to use non-fixed input shape to deal with different images. So I'm searching other solutions, mlx? libTorch? I don't know.
Dec ’24