cannot reproduce the fast mlmodel

[YingkunZhou]

I download the mlmodel which mentioned in Readme and also use almost the same environment to convert pytorch model to mlmodel

• torch==1.10.1
• coremltools==5.2.0
• timm==0.5.4

the left is mine, the right is downloaded from https://drive.google.com/file/d/1tZVxtbtAZoLLoDc5qqoUGulilksomLeK/view?usp=sharing

here is the layer distribution diff

and here is the performance diff

• Xcode version is Version 14.3 (14E222b)
• test on M1 mini

[Amshaker]

Hi,

Please change this boolean "CoreMLConversion" in swiftformer.py to True before converting your model to mlmodel. The normalization layer is currently not supported by coreml and this boolean disables this layer. I raised an issue into coreml and hopefully they support it soon.

Please let me know if you still not be able to reproduce the same performance.

Best regards,
Abdelrahman.

[YingkunZhou]

oh I see, thanks.