使用transformers.Trainer
时,其便利性与配置挑战性并存,形成了一种复杂的关系。引入了wandb作为内置功能的决定,可能源自某位营销天才的奇思妙想。虽然wandb在某些情况下确实提供了极大的便利,但在很多情况下,它的存在却显得多余。本篇文章旨在探讨如何应对这一需求。
文中内容包括:
- 如何在Python脚本中配置环境变量
- 如何通过
TrainingArguments
传递参数
撰写本文时参考了以下方法:
- 在Python脚本开头添加以下代码可以禁用wandb:
import os
os.environ["WANDB_DISABLED"] = "true"
但这种方法可能会触发一条警告信息,告诉你WANDB_DISABLED
环境变量即将弃用,并建议使用--report_to
标志来控制日志记录的集成方式。
- 另一种方法是在
TrainingArguments
中直接设置report_to="none"
,这也是警告信息中推荐的做法。