신경망 훈련에 PyTorch를 사용하는 방법-파이썬 튜토리얼-php.cn

신경망 훈련에 PyTorch를 사용하는 방법

소개:
PyTorch는 Python 기반의 오픈 소스 기계 학습 프레임워크로 유연성과 단순성으로 인해 많은 연구원과 엔지니어가 가장 먼저 선택합니다. 이 기사에서는 신경망 훈련에 PyTorch를 사용하는 방법을 소개하고 해당 코드 예제를 제공합니다.

1. PyTorch 설치
시작하기 전에 먼저 PyTorch를 설치해야 합니다. 공식 홈페이지(https://pytorch.org/)에서 제공하는 설치 가이드를 통해 운영체제 및 하드웨어에 적합한 버전을 선택하여 설치할 수 있습니다. 설치가 완료되면 Python으로 PyTorch 라이브러리를 가져오고 코드 작성을 시작할 수 있습니다.

2. 신경망 모델 구축
PyTorch를 사용하여 신경망을 훈련시키기 전에 먼저 적합한 모델을 구축해야 합니다. PyTorch는 자신만의 신경망 모델을 정의하기 위해 상속할 수 있는 torch.nn.Module이라는 클래스를 제공합니다. torch.nn.Module的类，您可以通过继承该类来定义自己的神经网络模型。

下面是一个简单的例子，展示了如何使用PyTorch构建一个包含两个全连接层的神经网络模型：

import torch
import torch.nn as nn

class Net(nn.Module):
    def __init__(self):
        super(Net, self).__init__()
        self.fc1 = nn.Linear(in_features=784, out_features=256)
        self.fc2 = nn.Linear(in_features=256, out_features=10)
    
    def forward(self, x):
        x = x.view(x.size(0), -1)
        x = self.fc1(x)
        x = torch.relu(x)
        x = self.fc2(x)
        return x

net = Net()

로그인 후 복사

在上面的代码中，我们首先定义了一个名为Net的类，并继承了torch.nn.Module类。在__init__方法中，我们定义了两个全连接层fc1和fc2。然后，我们通过forward方法定义了数据在模型中前向传播的过程。最后，我们创建了一个Net的实例。

三、定义损失函数和优化器
在进行训练之前，我们需要定义损失函数和优化器。PyTorch提供了丰富的损失函数和优化器的选择，可以根据具体情况进行选择。

下面是一个示例，展示了如何定义一个使用交叉熵损失函数和随机梯度下降优化器的训练过程：

loss_fn = nn.CrossEntropyLoss()
optimizer = torch.optim.SGD(net.parameters(), lr=0.01)

로그인 후 복사

在上面的代码中，我们将交叉熵损失函数和随机梯度下降优化器分别赋值给了loss_fn和optimizer变量。net.parameters()表示我们要优化神经网络模型中的所有可学习参数，lr参数表示学习率。

四、准备数据集
在进行神经网络训练之前，我们需要准备好训练数据集和测试数据集。PyTorch提供了一些实用的工具类，可以帮助我们加载和预处理数据集。

下面是一个示例，展示了如何加载MNIST手写数字数据集并进行预处理：

import torchvision
import torchvision.transforms as transforms

transform = transforms.Compose([
    transforms.ToTensor(),
    transforms.Normalize((0.5,), (0.5,)),
])

train_set = torchvision.datasets.MNIST(root='./data', train=True, download=True, transform=transform)
train_loader = torch.utils.data.DataLoader(train_set, batch_size=32, shuffle=True)

test_set = torchvision.datasets.MNIST(root='./data', train=False, download=True, transform=transform)
test_loader = torch.utils.data.DataLoader(test_set, batch_size=32, shuffle=False)

로그인 후 복사

在上面的代码中，我们首先定义了一个transform变量，用于对数据进行预处理。然后，我们使用torchvision.datasets.MNIST类加载MNIST数据集，并使用train=True和train=False参数指定了训练数据集和测试数据集。最后，我们使用torch.utils.data.DataLoader类将数据集转换成一个可以迭代的数据加载器。

五、开始训练
准备好数据集后，我们就可以开始进行神经网络的训练。在一个训练循环中，我们需要依次完成以下步骤：将输入数据输入到模型中，计算损失函数，反向传播更新梯度，优化模型。

下面是一个示例，展示了如何使用PyTorch进行神经网络训练：

for epoch in range(epochs):
    running_loss = 0.0
    for i, data in enumerate(train_loader):
        inputs, labels = data
        
        optimizer.zero_grad()
        
        outputs = net(inputs)
        loss = loss_fn(outputs, labels)
        
        loss.backward()
        optimizer.step()
        
        running_loss += loss.item()
        
        if (i+1) % 100 == 0:
            print('[%d, %5d] loss: %.3f' % (epoch+1, i+1, running_loss/100))
            running_loss = 0.0

로그인 후 복사

在上面的代码中，我们首先使用enumerate函数遍历了训练数据加载器，得到了输入数据和标签。然后，我们将梯度清零，将输入数据输入到模型中，计算预测结果和损失函数。接着，我们通过backward方法计算梯度，再通过step方法更新模型参数。最后，我们累加损失，并根据需要进行打印。

六、测试模型
训练完成后，我们还需要测试模型的性能。我们可以通过计算模型在测试数据集上的准确率来评估模型的性能。

下面是一个示例，展示了如何使用PyTorch测试模型的准确率：

correct = 0
total = 0

with torch.no_grad():
    for data in test_loader:
        inputs, labels = data
        outputs = net(inputs)
        _, predicted = torch.max(outputs.data, 1)
        total += labels.size(0)
        correct += (predicted == labels).sum().item()

accuracy = 100 * correct / total
print('Accuracy: %.2f %%' % accuracy)

로그인 후 복사

在上面的代码中，我们首先定义了两个变量correct和total，用于计算正确分类的样本和总样本数。接着，我们使用torch.no_grad()

다음은 PyTorch를 사용하여 두 개의 완전히 연결된 레이어가 포함된 신경망 모델을 구축하는 방법을 보여주는 간단한 예입니다.

rrreee
위 코드에서는 먼저 Net이라는 클래스를 정의하고 torch.nn에서 상속합니다. 모듈 클래스. __init__ 메서드에서는 두 개의 완전히 연결된 레이어 fc1 및 fc2를 정의합니다. 그런 다음 forward 메서드를 통해 모델에서 데이터의 순방향 전파 프로세스를 정의합니다. 마지막으로 Net 인스턴스를 만듭니다.

3. 손실 함수와 옵티마이저 정의

훈련 전에 손실 함수와 옵티마이저를 정의해야 합니다. PyTorch는 특정 상황에 따라 선택할 수 있는 다양한 손실 함수 및 최적화 프로그램을 제공합니다.

다음은 교차 엔트로피 손실 함수와 확률적 경사하강법 최적화 도구를 사용하여 훈련 과정을 정의하는 방법을 보여주는 예입니다.
위 코드에서는 교차 엔트로피 손실 함수와 확률적 경사하강법 최적화 도구를 할당합니다. 별도로 loss_fn 및 optimizer 변수가 제공됩니다. net.parameters()는 신경망 모델에서 학습 가능한 모든 매개변수를 최적화하려고 함을 나타내고 lr 매개변수는 학습 속도를 나타냅니다.

4. 데이터 세트 준비🎜 신경망을 훈련시키기 전에 훈련 데이터 세트와 테스트 데이터 세트를 준비해야 합니다. PyTorch는 데이터 세트를 로드하고 전처리하는 데 도움이 되는 몇 가지 실용적인 도구 클래스를 제공합니다. 🎜🎜다음은 MNIST 필기 숫자 데이터 세트를 로드하고 전처리하는 방법을 보여주는 예입니다. 🎜rrreee🎜위 코드에서는 먼저 transform 변수를 정의하여 데이터 전처리를 변환합니다. 그런 다음 torchvision.datasets.MNIST 클래스를 사용하여 MNIST 데이터세트를 로드하고 train=True 및 train=False 매개변수를 사용하여 훈련 데이터를 지정했습니다. 데이터 세트를 설정하고 테스트합니다. 마지막으로 torch.utils.data.DataLoader 클래스를 사용하여 데이터 세트를 반복 가능한 데이터 로더로 변환합니다. 🎜🎜5. 훈련 시작🎜 데이터 세트를 준비한 후 신경망 훈련을 시작할 수 있습니다. 훈련 루프에서는 입력 데이터를 모델에 입력하고, 손실 함수를 계산하고, 업데이트된 기울기를 역전파하고, 모델을 최적화하는 단계를 순서대로 완료해야 합니다. 🎜🎜다음은 신경망 훈련에 PyTorch를 사용하는 방법을 보여주는 예입니다. 🎜rrreee🎜위 코드에서는 먼저 enumerate 함수를 사용하여 훈련 데이터 로더를 탐색하여 입력 데이터와 레이블을 가져옵니다. 그런 다음 기울기를 0으로 만들고 입력 데이터를 모델에 공급한 다음 예측 및 손실 함수를 계산합니다. 다음으로 backward 메서드를 통해 기울기를 계산한 다음 step 메서드를 통해 모델 매개변수를 업데이트합니다. 마지막으로 손실을 누적하고 필요에 따라 인쇄합니다. 🎜🎜 6. 모델 테스트 🎜훈련이 완료된 후에도 모델의 성능을 테스트해야 합니다. 테스트 데이터 세트에 대한 정확도를 계산하여 모델의 성능을 평가할 수 있습니다. 🎜🎜다음은 PyTorch를 사용하여 모델의 정확성을 테스트하는 방법을 보여주는 예입니다. 🎜rrreee🎜위 코드에서는 먼저 corright 및 total 두 변수를 정의하고 사용합니다. 올바르게 분류된 샘플 수와 전체 샘플 수를 계산합니다. 다음으로 torch.no_grad() 컨텍스트 관리자를 사용하여 기울기 계산을 꺼서 메모리 소비를 줄입니다. 그런 다음 예측 결과를 순차적으로 계산하고 올바르게 분류된 샘플 수와 총 샘플 수를 업데이트합니다. 마지막으로 정확하게 분류된 샘플 수와 전체 샘플 수를 기준으로 정확도를 계산하여 인쇄합니다. 🎜🎜요약: 🎜이 글의 서론을 통해 신경망 훈련에 PyTorch를 사용하는 방법의 기본 단계를 이해했으며, 신경망 모델 구축, 손실 함수 및 옵티마이저 정의, 데이터 세트 준비, 훈련 시작 방법을 배웠습니다. 그리고 모델을 테스트합니다. 이 기사가 신경망 훈련에 PyTorch를 사용하는 작업과 연구에 도움이 되기를 바랍니다. 🎜🎜참고자료: 🎜🎜🎜PyTorch 공식 웹사이트: https://pytorch.org/🎜🎜PyTorch 문서: https://pytorch.org/docs/stable/index.html🎜🎜

위 내용은 신경망 훈련에 PyTorch를 사용하는 방법의 상세 내용입니다. 자세한 내용은 PHP 중국어 웹사이트의 기타 관련 기사를 참조하세요!