Huggingface——tensorboard监控训练过程

安装依赖

pip install tensorboard tensorboardX 
1

Trainer

• report_to：传入参数可以指定log工具，具体用法如下：（更多内容可参考文档：[Doc]）

from transformers import TrainingArguments, Trainer

# other imports and code

training_args = TrainingArguments(
    # other training arguments
    report_to=["tensorboard"],
)

trainer = Trainer(
    # other trainer settings
    args=training_args,
)

1
2
3
4
5
6
7
8
9
10
11
12
13
14

Vscode可视化TensorBoard

• 下载TensorBorad插件即可。