如何在PyTorch Lightning中消除seed_everything()的INFO日志信息
解决PyTorch Lightning seed_everything()重复输出日志的问题
以下是几个可行的解决办法,按推荐优先级排序:
最直接:调用seed_everything时关闭verbose模式
seed_everything本身提供了verbose参数,设为False就能直接关闭该日志输出,无需修改全局日志配置:from pytorch_lightning import seed_everything seed_everything(1234, verbose=False)如果是Trainer自动触发的seed设置
若你是通过Trainer的seed_everything参数自动设置种子,可以通过seed_everything_kwargs传递关闭verbose的参数:from pytorch_lightning import Trainer trainer = Trainer( seed_everything=1234, seed_everything_kwargs={"verbose": False} )针对性调整日志模块级别
如果你需要保留其他PL的INFO日志,仅屏蔽seed相关的输出,可以单独调整pytorch_lightning.utilities.seed模块的日志级别:import logging # 将该模块日志级别设为WARNING,INFO及以下日志会被屏蔽 logging.getLogger('pytorch_lightning.utilities.seed').setLevel(logging.WARNING)
内容的提问来源于stack exchange,提问作者jas0n
相关产品推荐
相关产品推荐

