如何在Transformers Trainer运行时禁用Neptune回调？

有几种可行方式能在单次运行中禁用Neptune回调，无需卸载依赖：

方法1：显式指定回调列表 #

初始化Trainer时，手动传入callbacks参数，只保留你需要的回调，排除Neptune相关回调。示例代码：

from transformers import Trainer, TrainingArguments
from transformers.trainer_callback import ProgressCallback

training_args = TrainingArguments(
    output_dir="./results",
    # 补充其他训练参数
)

# 仅添加需要的回调，比如进度条回调，不包含NeptuneCallback
trainer = Trainer(
    model=model,
    args=training_args,
    train_dataset=train_dataset,
    eval_dataset=eval_dataset,
    callbacks=[ProgressCallback()]
)

trainer.train()

*如果不需要任何额外回调，直接传callbacks=[]即可。

方法2：通过环境变量临时禁用 #

在脚本开头设置环境变量NEPTUNE_DISABLE=1，阻止Neptune回调自动初始化：

import os
os.environ["NEPTUNE_DISABLE"] = "1"

这种方式会全局临时禁用Neptune的自动集成，适合单次运行场景。

方法3：移除自动注册的回调 #

Transformers会自动注册Neptune回调，可在脚本开头手动移除：

from transformers.trainer_callback import CallbackHandler

# 遍历默认回调列表，移除NeptuneCallback
for callback in list(CallbackHandler._default_callbacks):
    if callback.__class__.__name__ == "NeptuneCallback":
        CallbackHandler._default_callbacks.remove(callback)

修改后后续创建的所有Trainer实例都不会包含Neptune回调，适合单次运行中多次创建Trainer的场景。

内容的提问来源于stack exchange，提问作者Poe Dator