Huggingface的Trainer类无法将日志写到logging文件中_huggingface的trainer怎么记录日志

作者：凡人多烦事01 | 2024-04-05 09:31:22

踩

huggingface的trainer怎么记录日志

在训练时想把这个信息也写到logging文件中

在training_args中设置参数：


training_args = transformers.TrainingArguments(
    logging_steps=2,
    eval_steps=2,
    logging_dir='logs/training.log',
    logging_strategy='steps',
    logging_first_step=True
)

在trainer.train()后面设置：


trainer.train()
 
for obj in trainer.state.log_history:
    logging.info(str(obj))

就可打印日志到日志文件中

声明：本文内容由网友自发贡献，不代表【wpsshop博客】立场，版权归原作者所有，本站不承担相应法律责任。如您发现有侵权的内容，请联系我们。转载请注明出处：https://www.wpsshop.cn/w/凡人多烦事01/article/detail/364815