architecture - Pytorch cifar 数据集:RuntimeError:大小不匹配,m1:[4 x 2048],m2:[1568 x 10]
问题描述
我正在使用 2 个基本层制作自己的网络,并在 CIFAR10 数据集上进行训练。我收到不匹配错误。
import torch
import torchvision
import torchvision.transforms as transforms
from torch.utils.data import Dataset, DataLoader
import torch.nn as nn
device = torch.device('cuda:0' if torch.cuda.is_available() else 'cpu')
num_epochs = 5
num_classes = 10
batch_size = 4
learning_rate = 0.001
print("---------Train Dataset----------")
train_dataset = torchvision.datasets.CIFAR10(root='../data/', train=True,
transform=transforms.ToTensor(), download=True)
print("---------Train Loader----------")
train_loader = torch.utils.data.DataLoader(dataset=train_dataset, batch_size=batch_size, shuffle=True)
class ConvNet(nn.Module):
def __init__(self, num_classes=10):
super(ConvNet, self).__init__()
# First Layer
self.layer1 = nn.Sequential(
nn.Conv2d(3, 16, kernel_size=3, stride=1, padding=1),
nn.BatchNorm2d(16),
nn.ReLU(),
nn.MaxPool2d(kernel_size=2, stride=2))
# Second Layer
self.layer2 = nn.Sequential(
nn.Conv2d(16, 32, kernel_size=3, stride=1, padding=1),
nn.BatchNorm2d(32),
nn.ReLU(),
nn.MaxPool2d(kernel_size=2, stride=2))
# Fully connected Layer
self.fc = nn.Linear(7 * 7 * 32, num_classes)
def forward(self, x):
out = self.layer1(x)
out = self.layer2(out)
out = out.reshape(out.size(0), -1)
out = self.fc(out)
return out
model = ConvNet(num_classes).to(device)
criterion = nn.CrossEntropyLoss()
optimizer = torch.optim.Adam(model.parameters(), lr=learning_rate)
print("--------Train Model----------")
total_step = len(train_loader)
print("Total Step", total_step)
for epoch in range(num_epochs):
for i, (images, labels) in enumerate(train_loader):
images = images.to(device)
labels = labels.to(device)
print("-----Forward Pass-------")
outputs = model(images)
loss = criterion(outputs, labels)
print ("----Backward Pass-------")
optimizer.zero_grad()
loss.backward()
optimizer.step()
if (i+1) % 100 == 0:
print(epoch+1, num_epochs, i+1, total_step, loss.item())
解决方案
推荐阅读
- javascript - nodejs没有连接到mongodb云服务器
- javascript - Redux 错误:您当前正在使用 NODE_ENV 之外的缩小代码
- arraylist - ArrayList 的深拷贝
- vue.js - 如果未填写强制输入,则在提交时提醒用户
- java - 如何在 try/catch 块上遵循 IntelliJ-IDEA 的建议“catch 分支相同”?
- node.js - 即使表单验证失败,文件仍会使用 multer 上传
- javascript - 事件处理程序上的 Reactjs 函数绑定
- android-studio - ConstraintSet 中的“toStartOf”和“toTopOf”等价物是什么?
- c# - Npgsql.PostgresException:'42P01:关系“表”不存在'
- django - 有什么方法可以更改用户身份验证登录页面中的用户名字段标签?