目录
需要的 SqueezeExcite代码,不是SqueezeExcite_o
依赖项:timm库。
代码出处:
GitHub - THU-MIG/RepViT: RepViT: Revisiting Mobile CNN From ViT Perspective
pip install timm==0.5.4
cuda版1060显卡运行时间 10ms左右
time 0.022658824920654297
time 0.013003349304199219
time 0.010505437850952148
time 0.013999462127685547
time 0.013000249862670898
time 0.