Pytorch 02 一个简单的分类器

最新推荐文章于 2023-04-21 11:46:47 发布

六甲横宝

最新推荐文章于 2023-04-21 11:46:47 发布

阅读量561

点赞数 1

CC 4.0 BY-SA版权

分类专栏： Pytorch 文章标签： Pytorch 深度学习

本文链接：https://blog.youkuaiyun.com/liujia2115/article/details/97496818

使用PyTorch训练了一个分类器，经过12000次迭代后，训练完成。网络在10000张测试图像上的准确率为56%，其中对ship的分类精度最高，达到77%。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

import torch
import torchvision
import torchvision.transforms as transforms
import matplotlib.pyplot as plt
import numpy as np
import torch.nn as nn
import torch.nn.functional as F
import torch.optim as optim


# 加载CIFAR10数据
transform = transforms.Compose(
    [transforms.ToTensor(),
     transforms.Normalize((0.5, 0.5, 0.5), (0.5, 0.5, 0.5))])

trainset = torchvision.datasets.CIFAR10(root='./data', train=True,
                                        download=True, transform=transform)
trainloader = torch.utils.data.DataLoader(trainset, batch_size=4,
                                          shuffle=True, num_workers=2)

testset = torchvision.datasets.CIFAR10(root='./data', train=False,
                                       download=True, transform=transform)
testloader = torch.utils.data.DataLoader(testset, batch_size=4,
                                         shuffle=False, num_workers=2)

classes = ('plane', 'car', 'bird', 'cat',