如何摆脱 PyTorch 闪电中的 INFO 日志记录消息

Question

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

Global seed set to 1234

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

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

谢谢

Answer 1

这对我有用：

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