pytorch-15-Train_Val_Test&交叉验证

最新推荐文章于 2025-09-06 05:14:47 发布

原创

最新推荐文章于 2025-09-06 05:14:47 发布 · 720 阅读

2 ·

CC 4.0 BY-SA版权

文章标签：

#pytorch #深度学习

本文详细探讨了使用PyTorch进行深度学习模型的训练、验证和测试流程，包括数据集划分、交叉验证的概念及其在PyTorch中的实现，旨在提升模型的泛化能力。

import  torch
import  torch.nn as nn
import  torch.nn.functional as F
import  torch.optim as optim
from    torchvision import datasets, transforms


batch_size=200
learning_rate=0.01
epochs=10

train_db = datasets.MNIST('./data', train=True, download=True,
                   transform=transforms.Compose([
                       transforms.ToTensor(),
                       transforms.Normalize((0.1307,), (0.3081,))
                   ]))
train_loader = torch.utils.data.DataLoader(
    train_db,
    batch_size=batch_size, shuffle=True)

test_db = datasets.MNIST('./data', train=False, transform=transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.1307,), (0.3081,))
]))
test_loader = torch.utils.data.D