• 2024-10-28基于pytorch的nonlocalblock
    论文《Non-localNeuralNetworks》为了满足即插即用的功能,本博客重写nonlocal块,并可以根据自己的喜好选择2D卷积或者3D卷积,并可以选择是否使用bn层或pool。nonlocalblock模块图3D代码如下:classNonlocal_3d(nn.Module):def__init__(self,dim,dim_inner,pool