2017-07-04 78 views
0

我正在學習CNN,我想用pytorch改變CNN模型與Wide Convolution,誰可以提供幫助?如何將CNN模型更改爲pytorch中的寬卷積?

self.conv23 = nn.Conv2d(Ci, len(Ks) * Co, (3, Co), padding=1) 

Traceback (most recent call last): 
    File "E:/workspace/pycharmworkspace/cnn-text-classification-pytorch-update/main.py", line 137, in <module> 
    train.train(train_iter, dev_iter, cnn, args) 
    File "E:\workspace\pycharmworkspace\cnn-text-classification-pytorch-update\train.py", line 40, in train 
    logit = model(feature) 
    File "C:\Users\bamtercelboo\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 206, in __call__ 
    result = self.forward(*input, **kwargs) 
    File "E:\workspace\pycharmworkspace\cnn-text-classification-pytorch-update\model.py", line 206, in forward 
    x21 = self.conv(x11, self.conv23) #(N,Co) 
    File "E:\workspace\pycharmworkspace\cnn-text-classification-pytorch-update\model.py", line 91, in conv 
    x = F.relu(conv(x)).squeeze(3) # (N,Co,W) 
    File "C:\Users\bamtercelboo\Anaconda3\lib\site-packages\torch\nn\modules\module.py", line 206, in __call__ 
    result = self.forward(*input, **kwargs) 
    File "C:\Users\bamtercelboo\Anaconda3\lib\site-packages\torch\nn\modules\conv.py", line 237, in forward 
    self.padding, self.dilation, self.groups) 
    File "C:\Users\bamtercelboo\Anaconda3\lib\site-packages\torch\nn\functional.py", line 43, in conv2d 
    return f(input, weight, bias) 
RuntimeError: kernel size should be greater than zero, but got kT: 3 kH: 200 kW: 0 at d:\downloads\pytorch-master-1\torch\lib\thnn\generic/VolumetricConvolutionMM.c:23 

回答

0

通常這是通過調整填充參數來完成的,但我可以看到你已經處理了這個。你能分享更多關於代碼的信息嗎?

PS對不起,評論答案。我不允許發表評論。

我想你可能需要的是調整Conv2D圖層中的組參數。

相關問題