用pytorch画loss曲线

当然可以，请参考以下代码：

import matplotlib.pyplot as plt
import numpy as np

# 定义训练轮数
epochs = 10

# 使用numpy生成随机的loss数据
train_losses = np.random.rand(epochs) * 10
valid_losses = np.random.rand(epochs) * 5

# 绘制曲线
plt.plot(range(epochs), train_losses, 'r', label='Training loss')
plt.plot(range(epochs), valid_losses, 'b', label='Validation loss')
plt.title('Loss over epochs')
plt.xlabel('Epochs')
plt.ylabel('Loss')
plt.legend()
plt.show()

其中，train_losses和valid_losses是分别代表训练集和验证集的loss数据，epochs是训练轮数。通过plt.plot函数可以绘制出对应的loss曲线，并使用plt.title、plt.xlabel等函数对图形进行美化。最终，调用plt.show函数将图形显示出来。