如何禁用PyTorch-闪电记录器的日志记录？

Question

PyTorch中的记录器-闪电打印关于要训练(或评估)的模型的信息，以及训练过程中的进度，

但是，在我的例子中，我想对记录器隐藏所有消息，以避免Jupyter Notebook中的输出泛滥。

我在正式文档页面https://pytorch-lightning.readthedocs.io/en/latest/common/trainer.html#trainer-flags上查看了训练器类的API，似乎没有任何选项可以关闭记录器的消息。

有一个参数log_every_n_steps，可以设置为大值，但是，日志记录结果后，显示了每个时代。

如何禁用日志记录？

Answer 1

我假设有两件事在淹水输出流方面特别困扰你：

第一，“权重摘要”：

  | 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闪电为关闭它们提供了非常清晰和优雅的解决方案：Trainer(progress_bar_refresh_rate=0)用于关闭进度条，Trainer(weights_summary=None)用于关闭重量汇总。

Answer 2

也许试着那样做？

Logging.getLogger(“包”).propagate= False

Answer 3

解决方案是将@Artyrm的建议和这里提出的答案结合起来，https://stackoverflow.com/a/52559560/13614416。

1. 获取所有pytorch_lightning记录器：

pl_loggers = logging.getLogger( name )表示名称在logging.root.manager.loggerDict中如果'pytorch_lightning‘在名称中

1. 将trainer.fit放在下面的构造中：

使用捕获的io.capture_output()：io.capture_output