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\Real-ESRGAN\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年03月28日 10:43:02
声明:本站所有文章,如无特殊说明或标注,均为本站原创发布。任何个人或组织,在未征得本站同意时,禁止复制、盗用、采集、发布本站内容到任何网站、书籍等各类媒体平台。如若本站内容侵犯了原著者的合法权益,可联系我们进行处理。
评论
56