在深度学习模型的训练中往往需要可视化loss曲线来对模型进行观察,以下是一个loss曲线可视化的代码:
# 可视化训练过程
fig1, ax1 = plt.subplots(figsize=(11, 8))
ax1.plot(range(0, epoch + 1, 1), Loss)
ax1.set_title("Loss of few-shot")
ax1.set_xlabel("Epoch")
ax1.set_ylabel("loss")
plt.savefig('few-shot-train_loss.png')
plt.clf()
plt.close()
放在每一轮训练结束之后使用,这里的Loss就是每一次训练完之后得到的平均loss,或者是自己设定的能代表一次训练之后的loss即可。