Huggingface的Trainer类无法将日志写到logging文件中

在训练时想把这个信息也写到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))

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

你可能感兴趣的:(人工智能)