HuggingFace 中的自定义日志记录

时间:2021-04-03 04:51:51

标签: huggingface-transformers

我正在运行 HuggingFace 官方存储库提供的 NER 教程(参见 here)。训练过程中打印以下信息

{'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() doc,但似乎唯一相关的选项如下,none 与日志记录的内容有关

  • logging_dir
  • logging_strategy
  • logging_first_step
  • logging_steps

0 个答案:

没有答案