2017-07-04 1 views
0

Ich lerne CNN, ich möchte CNN-Modell mit Wide Convolution von pytorch ändern, wer kann helfen?Wie ändert man CNN-Modell zu Wide Faltung in 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 

Antwort

0

Üblicherweise wird dies durch Einstellung der Polsterung Parameter getan, aber ich kann sehen Sie dies bereits behandelt. Kannst du mehr Informationen über den Code teilen?

PS Sorry für das Kommentieren in der Antwort. Ich darf noch keine Kommentare posten.

Ich denke, was Sie möglicherweise brauchen, ist die Anpassung der Gruppen Parameter in den Conv2D-Schichten.