首页 > 解决方案 > HuggingFace 中的自定义日志记录

问题描述

我正在运行 HuggingFace 官方 repo 提供的 NER 教程(请参阅此处)。在训练过程中打印以下信息

{'loss': 0.0724, 'learning_rate': 3.8425925925925924e-05, 'epoch': 0.69}        
{'loss': 0.0315, 'learning_rate': 2.6851851851851855e-05, 'epoch': 1.39}        
{'loss': 0.0198, 'learning_rate': 1.527777777777778e-05, 'epoch': 2.08}         
{'loss': 0.0147, 'learning_rate': 3.7037037037037037e-06, 'epoch': 2.78}        
{'train_runtime': 228.6028, 'train_samples_per_second': 9.449, 'epoch': 3.0} 

现在我还想在每个 epoch 之后打印出验证指标。我查看了TrainingArguments() 文档,但似乎唯一相关的选项如下,它们与日志记录的内容无关

标签: huggingface-transformers

解决方案


推荐阅读