如何在PyTorch闪电中摆脱INFO日志消息

Question

我正在使用PyTorch Lightning，我调用方法seed_everything()，但我不想看到INFO日志消息

Global seed set to 1234

在我的主算法的每次迭代。

我在PL对象的构造函数中尝试了logging.getLogger('pytorch_lightning').setLevel(logging.ERROR)，但它不起作用。我也尝试了这个答案中的建议，但它不起作用。

感谢

Answer 1

这对我有用:

import logging
log = logging.getLogger("pytorch_lightning")
log.propagate = False
log.setLevel(logging.ERROR)

Answer 2

我也找到了很多建议，最终我找到了一个更好的方法。虽然上述答案是有效的，但它将禁止输出其他信息。

在日志初始化前放置代码seed_everything()。就像这样:

import pytorch_lightning as pl
from pytorch_lightning import seed_everything
seed_everything(1234) 
logger = pl.loggers.TensorBoardLogger(save_dir="logs", name="example")