CoreML updatable Model with multiple Loss Functions

Hello,

I want to fine tune a CoreML model, with multiple binary outputs, on device. Therefore I would need multiple Loss Functions. If I try to compile that model I get the error:

Error Domain=com.apple.CoreML Code=3 "Error reading protobuf spec. validator error: This model has more than one loss layers specified, which is not supported at the moment." UserInfo={NSLocalizedDescription=Error reading protobuf spec. validator error: This model has more than one loss layers specified, which is not supported at the moment.}

Is it somehow possible to train an updatable CoreML model with multiple outputs on the device itself? I would really appreciate any help. Thank you!

Replies

CoreML on device training does not support multiple loss functions. Please file a feature request on feedbackassistant.apple.com.