PyTorch-Lightning 中的记录器打印有关要训练(或评估)的模型以及训练过程中的进度的信息,
但是,就我而言,我想隐藏记录器中的所有消息,以免淹没输出Jupyter Notebook
.
我在官方文档页面上查看了 Trainer 类的 APIhttps://pytorch-lightning.readthedocs.io/en/latest/common/trainer.html#trainer-flags似乎没有选项可以关闭记录器的消息。
有一个参数log_every_n_steps
可以设置为很大的值,但尽管如此,每个时期后的记录结果都会显示。
如何禁用日志记录?
我假设在泛洪输出流方面有两件事特别困扰您:
一、“体重总结”:
| Name | Type | Params
--------------------------------
0 | l1 | Linear | 100 K
1 | l2 | Linear | 1.3 K
--------------------------------
...
二、进度条:
Epoch 0: 74%|███████████ | 642/1874 [00:02<00:05, 233.59it/s, loss=0.85, v_num=wxln]
PyTorch Lightning 提供了非常清晰和优雅的解决方案来关闭它们:Trainer(progress_bar_refresh_rate=0)
用于关闭进度条和Trainer(weights_summary=None)
用于关闭体重摘要。
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系:hwhale#tublm.com(使用前将#替换为@)