(内含即插即用模块代码) ICCV 2023 RepViT | RepViTBlock最新开源移动端网络架构
本文会直接展示我们能用到的RepViTBlock这一即插即用的模块代码,有兴趣的可以先去看一看论文内容,论文链接:https://arxiv.org/abs/2307.09283下面我会直接放代码,以及我们的使用方法:m.bias.data.copy_(b)returnmclassResidual(torch.nn.Module):def__init__(self,m,drop=0.):super