使用transformers.Trainer时,其便利性与配置挑战性并存,形成了一种复杂的关系。引入了wandb作为内置功能的决定,可能源自某位营销天才的奇思妙想。虽然wandb在某些情况下确实提供了极大的便利,但在很多情况下,它的存在却显得多余。本篇文章旨在探讨如何应对这一需求。

文中内容包括:

  • 如何在Python脚本中配置环境变量
  • 如何通过TrainingArguments传递参数

撰写本文时参考了以下方法:

  • 在Python脚本开头添加以下代码可以禁用wandb:
import os
os.environ["WANDB_DISABLED"] = "true"

但这种方法可能会触发一条警告信息,告诉你WANDB_DISABLED环境变量即将弃用,并建议使用--report_to标志来控制日志记录的集成方式。

  • 另一种方法是在TrainingArguments中直接设置report_to="none",这也是警告信息中推荐的做法。
Logo

新一代开源开发者平台 GitCode,通过集成代码托管服务、代码仓库以及可信赖的开源组件库,让开发者可以在云端进行代码托管和开发。旨在为数千万中国开发者提供一个无缝且高效的云端环境,以支持学习、使用和贡献开源项目。

更多推荐