pytorch绘制loss曲线

PyTorch 可以通过 Matplotlib 库绘制 loss 曲线，具体实现方法如下：

1. 导入 Matplotlib 库：

   import matplotlib.pyplot as plt

2. 定义一个列表或数组来存储每个 epoch 的 loss 值：

   losses = [0.5, 0.4, 0.3, 0.2, 0.1]

3. 使用 Matplotlib 的 plot 函数绘制 loss 曲线：

   plt.plot(losses)
   plt.xlabel('Epoch')
   plt.ylabel('Loss')
   plt.show()

其中，`plt.plot(losses)` 用于绘制折线图，`plt.xlabel('Epoch')` 和 `plt.ylabel('Loss')` 用于设置坐标轴标签，`plt.show()` 用于显示图像。

通过上述代码，可以绘制出 loss 随 epoch 变化的曲线图，更好地了解模型训练过程中的损失值情况。