[英]Pytorch, TypeError: object() takes no parameters
這可能是一個初學者的問題,但是:盡管如此,當使用pytorch運行圖像分類器構建時,出現以下錯誤:
Traceback (most recent call last):
File "/pytorch/kanji_torch.py", line 47, in <module>
network = Network()
File "/pytorch/kanji_torch.py", line 113, in __init__
self.conv1 = nn.Conv2d(1, 32, 5)
File "/python3.5/site-packages/torch/nn/modules/conv.py", line 233, in __init__
False, _pair(0), groups, bias)
File "/python3.5/site-packages/torch/nn/modules/conv.py", line 32, in __init__
out_channels, in_channels // groups, *kernel_size))
TypeError: object() takes no parameters
我這樣定義網絡類:
class Network(torch.nn.Module):
def __init__(self):
super(Network, self).__init__()
self.conv1 = nn.Conv2d(1, 32, 5)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(32, 64, 5)
self.pool2 = nn.MaxPool2d(2, 2)
self.conv3 = nn.Conv2d(64, 64, 5)
self.pool2 = nn.MaxPool2d(2, 2)
self.fc1 = nn.Linear(64 * 5 * 5, 512)
self.fc2 = nn.Linear(512, 640)
self.fc3 = nn.Linear(640, 3756)
可以肯定的是,我正確地導入了所有相關的pytorch庫模塊。 (將torch.nn導入為nn,
進口火炬)
關於我在做什么錯的任何想法?
謝謝!
當我運行代碼時,您的pytorch版本可能有問題:
class Network(torch.nn.Module):
def __init__(self):
super(Network, self).__init__()
self.conv1 = nn.Conv2d(1, 32, 5)
self.pool = nn.MaxPool2d(2, 2)
self.conv2 = nn.Conv2d(32, 64, 5)
self.pool2 = nn.MaxPool2d(2, 2)
self.conv3 = nn.Conv2d(64, 64, 5)
self.pool2 = nn.MaxPool2d(2, 2)
self.fc1 = nn.Linear(64 * 5 * 5, 512)
self.fc2 = nn.Linear(512, 640)
self.fc3 = nn.Linear(640, 3756)
print(network)
輸出為:
Network (
(conv1): Conv2d(1, 32, kernel_size=(5, 5), stride=(1, 1))
(pool): MaxPool2d (size=(2, 2), stride=(2, 2), dilation=(1, 1))
(conv2): Conv2d(32, 64, kernel_size=(5, 5), stride=(1, 1))
(pool2): MaxPool2d (size=(2, 2), stride=(2, 2), dilation=(1, 1))
(conv3): Conv2d(64, 64, kernel_size=(5, 5), stride=(1, 1))
(fc1): Linear (1600 -> 512)
(fc2): Linear (512 -> 640)
(fc3): Linear (640 -> 3756)
)
我建議更新/重新安裝pytorch。
聲明:本站的技術帖子網頁,遵循CC BY-SA 4.0協議,如果您需要轉載,請注明本站網址或者原文地址。任何問題請咨詢:yoyou2525@163.com.