一、matplotlib库

在我们自己训练模型时，常常会使用matplotlib库来绘制oss和accuracy的曲线图，帮助我们分析模型的训练表现。
matplotlib库安装：pip install matplotlib

二、代码

import matplotlib.pyplot as plt
import torch
import torch.optim as optim  # 导入优化器模块#------------------------------------------------------------------#
#   定义损失函数
#------------------------------------------------------------------#
def loss_fn(y_true, y_pred):return torch.mean((y_true - y_pred)**2)#------------------------------------------------------------------#
#   定义模型
#------------------------------------------------------------------#
model = torch.nn.Linear(10, 1)#------------------------------------------------------------------#
#   定义训练，验证数据
#------------------------------------------------------------------#
x_train = torch.randn(1000, 10)
y_train = torch.randn(1000, 1)
x_val = torch.randn(1000, 10)
y_val = torch.randn(1000, 1)#------------------------------------------------------------------#
#   定义优化器
#------------------------------------------------------------------#
optimizer = optim.Adam(model.parameters(), lr=0.001)  # 使用 Adam 优化器，学习率为 0.001#------------------------------------------------------------------#
#   定义损失函数
#------------------------------------------------------------------#
train_loss_list = []
val_loss_list = []#------------------------------------------------------------------#
#   开始训练
#------------------------------------------------------------------#
for epoch in range(10000):# ------------------------------------------------------------------##   训练# ------------------------------------------------------------------## ------------------------------------------------------------------##   前向传播# ------------------------------------------------------------------#y_pred = model(x_train)# ------------------------------------------------------------------##   计算损失# ------------------------------------------------------------------#training_loss = loss_fn(y_train, y_pred)train_loss_list.append(training_loss.item())# ------------------------------------------------------------------##   反向传播# ------------------------------------------------------------------#training_loss.backward()# ------------------------------------------------------------------##   更新参数# ------------------------------------------------------------------#optimizer.step()# ------------------------------------------------------------------##   展示训练损失# ------------------------------------------------------------------#if epoch % 10 == 0:print(f"epoch {epoch}:training loss {training_loss.item()}")# ------------------------------------------------------------------##   验证# ------------------------------------------------------------------## ------------------------------------------------------------------##   前向传播# ------------------------------------------------------------------#y_pred = model(x_val)# ------------------------------------------------------------------##   计算损失# ------------------------------------------------------------------#val_loss = loss_fn(y_val, y_pred)val_loss_list.append(val_loss.item())# ------------------------------------------------------------------##   展示验证损失# ------------------------------------------------------------------#if epoch % 10 == 0:print(f"epoch {epoch}:validate loss {val_loss.item()}")# ------------------------------------------------------------------#
#   记录训练，验证损失
# ------------------------------------------------------------------#
plt.plot(train_loss_list,color="red",label="training_loss")
plt.plot(val_loss_list,color="blue",label="val_loss")
plt.xlabel("epoch")
plt.ylabel("loss")
plt.legend(loc='lower right')
plt.show()

运行结果

查看