epochs = list(range(1, num_epochs + 1))
train_losses = [] # 用于存储每个epoch的训练损失
plt.plot(epochs, train_losses, label='Training Loss')
plt.title('Training Loss Over Epochs')
plt.xlabel('Epochs')
plt.ylabel('Loss')
plt.legend()
plt.savefig('training_loss_plot.png')
把损失函数变成图片
发布时间 2023-11-10 09:57:24作者: 黑逍逍