Pytorch学习--神经网络--完整的模型训练套路

love_and_hope2024-11-12 22:52

一、下载数据集

python 复制代码 
train_data = torchvision.datasets.CIFAR10(root="datasets",train=True,transform=torchvision.transforms.ToTensor(),download=True)
train_data = torchvision.datasets.CIFAR10(root="datasets",train=False,transform=torchvision.transforms.ToTensor(),download=True)

二、数据大小

python 复制代码 
train_data_len = len(train_data)
test_data_len = len(test_data)
print(train_data_len)
print(test_data_len)

三、数据加载

python 复制代码 
train = DataLoader(train_data,batch_size=64)
test = DataLoader(test_data,batch_size=64)

四、模型定义

python 复制代码 
class Mary(nn.Module):
    def __init__(self):
        super(Mary,self).__init__()
        self.model1 = nn.Sequential(
            Conv2d(3, 32, 5, padding=2),
            MaxPool2d(2),
            Conv2d(32, 32, 5, padding=2),
            MaxPool2d(2),
            Conv2d(32, 64, 5, padding=2),
            MaxPool2d(2),
            Flatten(),
            Linear(1024, 64),
            Linear(64, 10)
        )
    def forward(self,x):
        x = self.model1(x)
        return x

五、定义对象

python 复制代码 
Yorelee = Mary()

六、损失函数

python 复制代码 
loss_fn = nn.CrossEntropyLoss()

七、优化器

python 复制代码 
learning_rate = 0.01
optimizer = torch.optim.SGD(Yorelee.parameters(),lr=learning_rate)

八、训练过程

python 复制代码 
epoch = 10
for i in range(epoch):
    print("***********第{}轮训练***********".format(i+1))
    index = 1
    for data in train:
        imgs,targets = data
        output = Yorelee(imgs)
        loss = loss_fn(output,targets)

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        print("({}) ".format(index),loss.item())
        index += 1

九、训练过程+测试过程

python 复制代码 
# 训练过程
epoch = 10
for i in range(epoch):
    print("***********第{}轮训练***********".format(i+1))
    index = 1
    for data in train:
        imgs,targets = data
        output = Yorelee(imgs)
        loss = loss_fn(output,targets)

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        print("({}) ".format(index),loss.item())
        index += 1

    #测试过程
    total_loss = 0
    with torch.no_grad():
        for data in test:
            imgs,targets = data
            output = Yorelee(imgs)
            loss = loss_fn(output,targets)
            total_loss += loss
    print("测试： ",total_loss.item())

十、Tensorboard 可视化

python 复制代码 
#  Tensorboard 可视化
writer = SummaryWriter("logs")

# 训练过程
epoch = 10
for i in range(epoch):
    print("***********第{}轮训练***********".format(i+1))
    index = 1
    for data in train:
        imgs,targets = data
        output = Yorelee(imgs)
        loss = loss_fn(output,targets)

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        print("({}) ".format(index),loss.item())
        if index % 100 == 0:
            writer.add_scalar("train_loss",loss,index)
        index += 1

    #测试过程
    total_loss = 0
    with torch.no_grad():
        for data in test:
            imgs,targets = data
            output = Yorelee(imgs)
            loss = loss_fn(output,targets)
            total_loss += loss
    print("测试： ",total_loss.item())
    writer.add_scalar("test_loss", total_loss, i)
writer.close()

十一、对于分类任务

可以用准确率来衡量，而不是用 total_loss 来衡量效果

先来举个简单的例子：


代码如下：

python 复制代码 
import torch

output = torch.tensor([[0.1,0.2],
                       [0.3,0.4]])
output = torch.argmax(output,dim=1)
preds = torch.tensor([0,1])
print((output==preds).sum())

输出：

python 复制代码 
tensor(1)

十二、对测试过程的重新改进

python 复制代码 
#  Tensorboard 可视化
writer = SummaryWriter("logs")

# 训练过程
epoch = 10
for i in range(epoch):
    print("***********第{}轮训练***********".format(i+1))
    index = 1
    for data in train:
        imgs,targets = data
        output = Yorelee(imgs)
        loss = loss_fn(output,targets)

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        print("({}) ".format(index),loss.item())
        if index % 100 == 0:
            writer.add_scalar("train_loss",loss,index)
        index += 1

    #测试过程
    total_loss = 0
    with torch.no_grad():
        find_true = 0
        for data in test:
            imgs,targets = data
            output = Yorelee(imgs)
            # loss = loss_fn(output,targets)
            # total_loss += loss
            output = torch.argmax(output,dim=1)
            find_true += (output == targets).sum()
    print("测试： ",find_true/test_data_len.item())
    writer.add_scalar("test_loss", find_true/test_data_len, i)

十三、总结

model.py

python 复制代码 
import torch
from torch import nn
from torch.nn import Conv2d, MaxPool2d, Flatten, Linear


class Mary(nn.Module):
    def __init__(self):
        super(Mary,self).__init__()
        self.model1 = nn.Sequential(
            Conv2d(3, 32, 5, padding=2),
            MaxPool2d(2),
            Conv2d(32, 32, 5, padding=2),
            MaxPool2d(2),
            Conv2d(32, 64, 5, padding=2),
            MaxPool2d(2),
            Flatten(),
            Linear(1024, 64),
            Linear(64, 10)
        )
    def forward(self,x):
        x = self.model1(x)
        return x

if __name__ == '__main__':
    input = torch.ones((64,3,32,32))
    Yorelee = Mary()
    output = Yorelee(input)
    print(output.shape)

train.py

python 复制代码 
import torch
import torchvision
from torch.utils.data import DataLoader
from torch.utils.tensorboard import SummaryWriter

from model import *

# 数据集
train_data = torchvision.datasets.CIFAR10(root="datasets",train=True,transform=torchvision.transforms.ToTensor(),download=True)
test_data = torchvision.datasets.CIFAR10(root="datasets",train=False,transform=torchvision.transforms.ToTensor(),download=True)

# 数据大小
train_data_len = len(train_data)
test_data_len = len(test_data)
print(train_data_len)
print(test_data_len)

# 数据加载
train = DataLoader(train_data,batch_size=64)
test = DataLoader(test_data,batch_size=64)

# 定义对象
Yorelee = Mary()

# 损失函数
loss_fn = nn.CrossEntropyLoss()

# 优化器
learning_rate = 0.01
optimizer = torch.optim.SGD(Yorelee.parameters(),lr=learning_rate)

#  Tensorboard 可视化
writer = SummaryWriter("logs")

# 训练过程
epoch = 10
for i in range(epoch):
    print("***********第{}轮训练***********".format(i+1))
    index = 1
    for data in train:
        imgs,targets = data
        output = Yorelee(imgs)
        loss = loss_fn(output,targets)

        optimizer.zero_grad()
        loss.backward()
        optimizer.step()
        print("({}) ".format(index),loss.item())
        if index % 100 == 0:
            writer.add_scalar("train_loss",loss,index)
        index += 1

    #测试过程
    total_loss = 0
    with torch.no_grad():
        find_true = 0
        for data in test:
            imgs,targets = data
            output = Yorelee(imgs)
            # loss = loss_fn(output,targets)
            # total_loss += loss
            output = torch.argmax(output,dim=1)
            find_true += (output == targets).sum()
    print("测试： ",find_true/test_data_len.item())
    writer.add_scalar("test_loss", find_true/test_data_len, i)
writer.close()

十四、XX.train() 和 XX.eval() 的说明



相关推荐
zskj_zhyl30 分钟前
家庭健康能量站：微高压氧舱结合艾灸机器人，智享双重养生SPA
人工智能·科技·安全·机器人
朗迪锋32 分钟前
数字孪生 ：提高制造生产力的智能方法
大数据·人工智能·制造
网安INF1 小时前
【论文阅读】-《HopSkipJumpAttack: A Query-Efficient Decision-Based Attack》
论文阅读·人工智能·深度学习·网络安全·对抗攻击
你知道网上冲浪吗1 小时前
【原创理论】Stochastic Coupled Dyadic System (SCDS)：一个用于两性关系动力学建模的随机耦合系统框架
python·算法·数学建模·数值分析
钢铁男儿1 小时前
Python 正则表达式核心元字符全解析
python
yatingliu20191 小时前
HiveQL | 个人学习笔记
hive·笔记·sql·学习
武当豆豆1 小时前
C++编程学习（第25天）
开发语言·c++·学习
风和日丽 随波逐流1 小时前
java17学习笔记-Deprecate the Applet API for Removal
笔记·学习
杨荧2 小时前
基于Python的宠物服务管理系统 Python+Django+Vue.js
大数据·前端·vue.js·爬虫·python·信息可视化
淮北也生橘122 小时前
Linux的ALSA音频框架学习笔记
linux·笔记·学习
热门推荐
01UV安装并设置国内源02DeepSeek更新！速览DeepSeek V3.1新特性03Qwen3-Coder 快速上手教程 | Qwen Code + Claude Code04KGG转MP3工具|非KGM文件|解密音频05蜘蛛磁力 搜索引擎大全，如何使用蜘蛛磁力查找磁力链接06【2025.08.06最新版】Android Studio下载、安装及配置记录（自动下载sdk）072025最新国内服务器可用docker源仓库地址大全（2025年8月更新）08阿里开源首个图像生成基础模型——Qwen-Image本地部署教程，超强中文渲染能力刷新SOTA！09NVIDIA显卡驱动、CUDA、cuDNN 和 TensorRT 版本匹配指南10TRAE 规则（Rules）配置指南：个人习惯、团队规范与最佳实践