Andy comments

Results 2 comments of


                                            Andy

RuntimeError: Given groups=1, weight of size [128, 512, 1, 1], expected input[4, 2048, 1, 1] to have 512 channels, but got 2048 channels instead

> 我也是初学者，试着回答一下，不对的话请大神指正，nn.Conv2d第一个参数是输入通道数，这个输入通道数是不可改变的，是由你的图片通道数决定的但是代码里本来是这么写的： class _PSPHead(nn.Module): def init(self, nclass, norm_layer=nn.BatchNorm2d, norm_kwargs=None, **kwargs): super(_PSPHead, self).init() self.psp = _PyramidPooling(2048, norm_layer=norm_layer, norm_kwargs=norm_kwargs) self.block = nn.Sequential( nn.Conv2d(4096, 512, 3, padding=1, bias=False), norm_layer(512, **({} if norm_kwargs...

RuntimeError: Given groups=1, weight of size [128, 512, 1, 1], expected input[4, 2048, 1, 1] to have 512 channels, but got 2048 channels instead

> > > 我也是初学者，试着回答一下，不对的话请大神指正，nn.Conv2d第一个参数是输入通道数，这个输入通道数是不可改变的，是由你的图片通道数决定的 > > > > > > 但是代码里本来是这么写的： class _PSPHead(nn.Module): def init(self, nclass, norm_layer=nn.BatchNorm2d, norm_kwargs=None, **kwargs): super(_PSPHead, self).init() self.psp = _PyramidPooling(2048, norm_layer=norm_layer, norm_kwargs=norm_kwargs) self.block = nn.Sequential( nn.Conv2d(4096,...