• 欢迎大家交换友链,可在https://www.stubbornhuang.com/申请友情链接进行友链交换申请!

  • 本站由于前段时间遭受到大量临时和国外邮箱注册,所以对可注册的邮箱类型进行了限制!

  • 本站会放置Google广告用于维持域名以及网站服务器费用。

  • 在本站开通年度VIP,无限制下载本站资源和阅读本站文章

  • 如果觉得本站的内容有帮助,可以考虑打赏博主哦!

  • 计算机图形学与计算几何经典必备书单整理,下载链接可参考:https://www.stubbornhuang.com/1256/

  • 问题反馈可发送邮件到stubbornhuang@qq.com

  • 感谢大家访问本站,希望本站的内容可以帮助到大家!

  • 工资「喂饱肚子」,副业「养活灵魂」!

修改basicsr模型训练时的loss打印输出形式,从科学计数修改为小数点形式

图片超分 发布于2025-03-28 阅读 4,622次 0次评论 0次点赞 本文共1037个字,阅读需要3分钟。

1 修改basicsr模型训练时的loss打印输出形式,从科学计数修改为小数点形式

找到basicsr的basicsr/utils/logger.py文件,如果是以包的方式安装的basicsr,就在你的conda环境中找到basicsr的包所在位置,然后再找到basicsr/utils/logger.py文件。

一般在conda虚拟环境中,basicsr包的安装路径为:

  • Ubuntu上如下:你的conda目录/envs/你的虚拟环境名称/lib/你的python版本/site-packages/basicsr/utils/logger.py
  • Windows上如下:你的conda目录\envs\你的虚拟环境名称\Lib\site-packages\basicsr\utils\logger.py

修改basicsr/utils/logger.py文件中的MessageLogger类的call函数,将该文件的

        # other items, especially losses
        for k, v in log_vars.items():
            message += f'{k}: {v:.4e} '
            # tensorboard logger
            if self.use_tb_logger and 'debug' not in self.exp_name:
                if k.startswith('l_'):
                    self.tb_logger.add_scalar(f'losses/{k}', v, current_iter)
                else:
                    self.tb_logger.add_scalar(k, v, current_iter)
        self.logger.info(message)

修改为

        # other items, especially losses
        for k, v in log_vars.items():
            message += f'{k}: {v:.6f} ' # 保留6位小数点
            # tensorboard logger
            if self.use_tb_logger and 'debug' not in self.exp_name:
                if k.startswith('l_'):
                    self.tb_logger.add_scalar(f'losses/{k}', v, current_iter)
                else:
                    self.tb_logger.add_scalar(k, v, current_iter)
        self.logger.info(message)

即可。
这样在模型训练时,各个loss项的输出就不是科学计数形式,而是小数点的形式。

欢迎扫码关注我的微信公众号,及时获取文章更新

微信公众号二维码

本文作者:StubbornHuang

版权声明:本文为站长原创文章,如果转载请注明原文链接!

原文标题:修改basicsr模型训练时的loss打印输出形式,从科学计数修改为小数点形式

原文链接:https://www.stubbornhuang.com/3149/

发布于:2025年03月28日 10:42:23

修改于:2025年04月08日 11:06:46

声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。

文章末尾
上一篇
ubuntu服务器配置v2ray客户端
建站运维
下一篇
修改basicsr模型训练的流程,在模型训练和微调之前就验证一遍数据集
图片超分
当前分类随机文章推荐

发表评论

您必须 [ 登录 ] 才能发表留言!

关注我们的公众号

微信公众号