search

kaijieshi7 / Dynamic-convolution-Pytorch

Pytorch!!!Pytorch!!!Pytorch!!! Dynamic Convolution: Attention over Convolution Kernels (CVPR-2020)
553 stars 91 forks source link

The initialization of bias in Dynamic_convNd is undefined #8

Open kjohew opened 3 years ago

kjohew commented 3 years ago

self.bias = nn.Parameter(torch.Tensor(K, out_planes)) should be changed to self.bias = nn.Parameter(torch.zeros(K, out_planes))

kaijieshi7 commented 3 years ago

Thank you。