pytorch画loss曲线的实例

很高兴为您解答。以下是 PyTorch 画 loss 曲线的实例代码：

import torch
import matplotlib.pyplot as plt

# 假设 loss 存储在一个列表中
losses = [0.1, 0.08, 0.06, 0.04, 0.02, 0.01]

# 画出 loss 曲线
plt.plot(losses)
plt.xlabel('Epoch')
plt.ylabel('Loss')
plt.title('Loss Curve')
plt.show()

上述代码可以让您快速画出 loss 曲线，以便更直观地了解训练的效果。希望对您有所帮助。