loss曲线绘制

在深度学习模型的训练中往往需要可视化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即可。

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值