数据可视化与大模型训练的实践探索

import matplotlib.pyplot as plt

# 假设这是训练过程中记录下来的损失值

losses = [0.5, 0.4, 0.3, 0.25, 0.23, 0.22]

# 绘制损失曲线

plt.plot(losses)

plt.xlabel('Epoch')

plt.ylabel('Loss')

plt.title('Training Loss Over Epochs')

plt.show()

]]>

val_losses = [0.6, 0.5, 0.45, 0.4, 0.38, 0.37]

# 同时绘制两条曲线

plt.plot(losses, label='Train Loss')

plt.plot(val_losses, label='Validation Loss')

plt.xlabel('Epoch')

plt.ylabel('Loss')

plt.title('Train vs Validation Loss')

plt.legend()

plt.show()

]]>