THU-MIG / RepViT

RepViT: Revisiting Mobile CNN From ViT Perspective [CVPR 2024] and RepViT-SAM: Towards Real-Time Segmenting Anything
https://arxiv.org/abs/2307.09283
Apache License 2.0
756 stars 56 forks source link

how to use fuse function in model #22

Closed XieZixiUSTC closed 10 months ago

jameslahm commented 10 months ago

Thanks for your interest. You could directly call utils.replace_batchnorm(model) to invoke the fuse function in model.