構(gòu)建yoloV3的resblock時，python的一個小坑

在構(gòu)建一個簡單的卷積層時，出現(xiàn)如下的錯誤：

代碼如下：

class resblock(nn.Module):def __init__(self,inputDims):super(resblock,self).__init__()self.inputChannels = inputDimsself.middleChannels = int(inputDims/2)#報錯行self.conv1 = nn.Conv2d(in_channels=inputDims,out_channels=inputDims/2,kernel_size=1,stride=1)self.bn1 = nn.BatchNorm2d(self.middleChannels)self.active1 = nn.LeakyReLU()self.conv3x3 = nn.Conv2d(in_channels=self.middleChannels,out_channels=inputDims,kernel_size=3,stride=1,padding=1)self.bn2 = nn.BatchNorm2d(inputDims)self.active2 = nn.LeakyReLU()self.shortcut = shortCut()def forward(self,x):y1 = self.active1(self.bn1(self.conv1(x)))y = self.active2(self.bn2(self.conv3x3(y1)))return self.shortcut(x,y)

產(chǎn)生的原因是因?yàn)樵趐ython3中兩個整型相除得到的是浮點(diǎn)型，例如：4/2=2.0,而在構(gòu)建卷積時的參數(shù)要求時整型

總結(jié)

以上是生活随笔為你收集整理的【深度学习】empty() received an invalid combination of arguments - got (tuple, dtype=NoneType, device=None的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。