Huggingface——tensorboard监控训练过程

最新推荐文章于 2025-04-21 19:15:28 发布

Irving.Gao

最新推荐文章于 2025-04-21 19:15:28 发布

阅读量2.1k

点赞数

分类专栏： LLM 文章标签： python

本文链接：https://blog.csdn.net/qq_45779334/article/details/134465640

版权

LLM 专栏收录该内容

3 篇文章

订阅专栏

本文介绍了如何在Python中使用pip安装TensorBoard和tensorboardX，以及如何通过TrainingArguments和Trainer设置将训练日志报告给TensorBoard。同时，提到了在VSCode中安装并可视化TensorBoard的方法。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

安装依赖

pip install tensorboard tensorboardX

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,
)