add_scalar和add_scalars函数

add_scalar和add_scalars

这两个函数是用来保存程序中的数据,然后利用tensorboard工具来进行可视化的。

add_scalar

 sw.add_scalar('lr'   , optimizer.param_groups[0]['lr'], global_step=global_step)

这行代码的含义就是利用add_scalar函数来生成一个“lr”名称的二维函数,横轴X值为“optimizer.param_groups[0][‘lr’]”,纵轴Y值为global_step。

即:

    • 7
      点赞
    • 16
      收藏
      觉得还不错? 一键收藏
    • 3
      评论
    绘制损失函数曲线可以使用两种方法,一种是使用TensorBoard,另一种是使用matplotlib。 使用TensorBoard的方法如下: 首先,需要导入必要的库和函数: ```python from torch.utils.tensorboard import SummaryWriter import loadFromPickle import drwaLossFunction ``` 接下来,创建pickle文件并获取相关信息: ```python drwaLossFunction.createPickleFiles() info = loadFromPickle.getInfo() ``` 然后,使用SummaryWriter创建一个writer对象,指定保存路径: ```python writer = SummaryWriter('绘制损失函数曲线/单独绘制') ``` 使用writer对象添加scalar数据,可以通过zip(*info)将epoch、置信度损失和定位损失分别对应起来: ```python for epoch, conf, loc in zip(*info): writer.add_scalar('置信度损失', conf, epoch) writer.add_scalar('定位损失', loc, epoch) ``` 关闭writer对象: ```python writer.close() ``` 如果需要进行三者对比,可以创建一个新的writer对象,并使用add_scalars函数添加多个scalar数据: ```python writer = SummaryWriter('绘制损失函数曲线/三者对比') for epoch, conf, loc in zip(*info): writer.add_scalars('损失函数', {'conf':conf, 'loc':loc, 'conf loc': conf loc}, epoch) writer.close() ``` 最后,使用命令行运行tensorboard来查看结果: ```python tensorboard --logdir=绘制损失函数曲线 ``` 使用matplotlib的方法如下: 首先,导入必要的库和函数: ```python import drwaLossFunction import loadFromPickle import matplotlib import matplotlib.pyplot as plt ``` 创建pickle文件并获取相关信息: ```python drwaLossFunction.createPickleFiles() info = loadFromPickle.getInfo() epoch, conf, loc = info ``` 使用plt.plot函数绘制定位损失和置信度损失曲线: ```python plt.plot(epoch, loc,'rx-.', label='定位损失') plt.plot(epoch, conf, 'go-', label='置信度损失') ``` 设置横纵轴的标签和图例位置: ```python plt.xlabel('训练世代: epoch') plt.ylabel('损失值: loss') plt.legend(loc='upper right', frameon=True) ``` 显示图形: ```python plt.show() ```

    “相关推荐”对你有帮助么?

    • 非常没帮助
    • 没帮助
    • 一般
    • 有帮助
    • 非常有帮助
    提交
    评论 3
    添加红包

    请填写红包祝福语或标题

    红包个数最小为10个

    红包金额最低5元

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

    抵扣说明:

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

    余额充值