阅读量:0
在PyTorch中，可以通过使用优化器来优化模型。PyTorch提供了许多不同的优化器，其中最常用的是SGD（随机梯度下降）和Adam优化器。
下面是一个简单的示例，展示如何在PyTorch中使用SGD优化器来优化模型：
import torch import torch.nn as nn import torch.optim as optim # 定义一个简单的神经网络模型 class SimpleModel(nn.Module): def __init__(self): super(SimpleModel, self).__init__() self.fc = nn.Linear(10, 1) def forward(self, x): return self.fc(x) # 实例化模型和优化器 model = SimpleModel() optimizer = optim.SGD(model.parameters(), lr=0.01) # 定义损失函数 criterion = nn.MSELoss() # 模拟输入数据和标签 inputs = torch.randn(1, 10) labels = torch.randn(1, 1) # 训练模型 optimizer.zero_grad() outputs = model(inputs) loss = criterion(outputs, labels) loss.backward() optimizer.step() 在上面的示例中，首先定义了一个简单的神经网络模型SimpleModel，然后实例化了一个SGD优化器，并定义了损失函数。接着模拟了输入数据和标签，并通过反向传播和优化器更新来训练模型。
除了SGD优化器外，PyTorch还提供了其他优化器，如Adam、Adagrad、RMSprop等，可以根据具体情况选择适合的优化器来优化模型。
