ml-cvnets icon indicating copy to clipboard operation
ml-cvnets copied to clipboard

add apple's new model: FastViT

Open rotem154154 opened this issue 2 years ago • 4 comments
trafficstars

Apple just released very exiting model named FastViT. is there any plan on relese it? https://arxiv.org/abs/2303.14189

rotem154154 avatar Mar 27 '23 12:03 rotem154154

Hi, I try to reproduce the FastViT. Here is the coreml model of my reproduced FastViT-T12. However, I find its mobile latency is much bigger than the latency data reported in Table 5 in the paper. I use iPhone 12 and Xcode coreml performance tool to measure the latency. According to Table 9 in the mobileone paper and the Table 5 in the FastViT paper, FastViT-T12 can obtain the similar mobile latency as MobileOne-S2, \ie, about 1.2ms. image image However, the actual latency I measured is MobileOne-S2: 1.13ms and FastViT-T12: 1.73ms. image image I wonder if there is something wrong with my reproduction. Could you give me some guide or would you mind sharing the code? Thanks a lot! Any advice is appreciated.

jameslahm avatar Jun 27 '23 14:06 jameslahm

@jameslahm can you link to repo?

vizaiapp avatar Jun 29 '23 20:06 vizaiapp

@jameslahm Can I see your source code?

shituo123456 avatar Jul 05 '23 02:07 shituo123456

It seems to be available now: https://github.com/apple/ml-cvnets/blob/c668ad13d67b3e8829e810d4b6e16ad7f77fd4aa/cvnets/models/classification/fastvit.py. But there is no pretrained model yet.

Wikidepia avatar Jul 12 '23 22:07 Wikidepia