我刚刚开始学习Pytork并创建我的第一个CNN。数据集包含3360个RGB图像,我将它们转换为[3360, 3, 224, 224]
张量。数据和标签位于dataset(torch.utils.data.TensorDataset)
中。以下是培训代码
def train_net():
dataset = ld.load()
data_iter = Data.DataLoader(dataset, batch_size=168, shuffle=True)
net = model.VGG_19()
summary(net, (3, 224, 224), device="cpu")
loss_func = nn.CrossEntropyLoss()
optimizer = optim.SGD(net.parameters(), lr=0.001, momentum=0.9, dampening=0.1)
scheduler = torch.optim.lr_scheduler.StepLR(optimizer, step_size=50, gamma=0.1)
for epoch in range(5):
print("epoch:", epoch + 1)
train_loss = 0
for i, data in enumerate(data_iter, 0):
x, y = data
print(x.dtype)
optimizer.zero_grad()
out = net(x)
loss = loss_func(out, y)
loss.backward()
optimizer.step()
train_loss += loss.item()
if i % 100 == 99:
print("loss:", train_loss / 100)
train_loss = 0.0
print("finish train")
那么我有一个错误:
Traceback (most recent call last):
File "D:/python/DeepLearning/VGG/train.py", line 52, in <module>
train_net()
File "D:/python/DeepLearning/VGG/train.py", line 29, in train_net
out = net(x)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\python\DeepLearning\VGG\model.py", line 37, in forward
out = self.conv3_64(x)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\python\lib\site-packages\torch\nn\modules\container.py", line 117, in forward
input = module(input)
File "D:\python\lib\site-packages\torch\nn\modules\module.py", line 727, in _call_impl
result = self.forward(*input, **kwargs)
File "D:\python\lib\site-packages\torch\nn\modules\conv.py", line 423, in forward
return self._conv_forward(input, self.weight)
File "D:\python\lib\site-packages\torch\nn\modules\conv.py", line 419, in _conv_forward
return F.conv2d(input, weight, self.bias, self.stride,
RuntimeError: expected scalar type Double but found Float
我认为x有问题,我通过print(x.dtype)
打印它的类型:
torch.float64
它是双精度的,而不是浮动的。你知道怎么了吗?谢谢你的帮助
该错误实际上是指conv层的权重,在调用矩阵乘法时,默认情况下,conv层位于
float32
。因为您的输入是double
(float64
在pytorch中),而conv中的权重是float
因此,您的解决方案是:
这肯定会奏效
相关问题 更多 >
编程相关推荐