conversion to coreml doesn't work

bes-dev / MobileStyleGAN.pytorch

An official implementation of MobileStyleGAN in PyTorch

Apache License 2.0

672 stars 81 forks source link

conversion to coreml doesn't work #41

Open rotem154154 opened 2 years ago

rotem154154 commented 2 years ago

when i try to convert to coreml the code crash with: RuntimeError: PyTorch convert function for op 'numel' not implemented.

bes-dev commented 2 years ago

@rotem154154 which versions of pytorch and corml-tools do you use?

rotem154154 commented 2 years ago

i tried with pytorch 1.12.0 and 1.13.0.dev20220726 both using coreml 5.2.0 and 5.0.0

bes-dev commented 2 years ago

@rotem154154 sorry for the late response. I checked model conversion to coreml and it works good for me with coremltools==5.2.0. Please, try to update coremltools in your python env.