如何在变压器训练器运行中禁用海王星回调？

Question

在为偶尔的ML实验日志安装Neptune.ai之后，默认情况下，它会被包括在所有transformers.trainer运行的回调列表中。因此，它需要使用令牌进行适当的初始化，否则会引发NeptuneMissingConfiguration错误，要求令牌和项目名称。这真的很烦人，我更希望海王星回调限制在警告上，或者在没有提供令牌的情况下让它被禁用。不幸的是，除了完全卸载Neptune.ai之外，没有明显的方法禁用此回调。类/回调上的doc页面指出，默认情况下启用了此回调，并且无法禁用它(与环境变量可以禁用的其他回调不同)。

问题:如何在每次运行的基础上禁用海王星回调？

Answer 1

要在变形金刚训练器运行中禁用海王星回调，可以将无海王星标志传递给trainer.train()函数。

trainer = Trainer(
    model=model,
    args=args,
    train_dataset=train_dataset,
    eval_dataset=eval_dataset,
    no_neptune=True
)
trainer.train()

Answer 2

显然，trainer初始化后的这段代码有助于：

for cb in trainer.callback_handler.callbacks:
    if isinstance(cb, transformers.integrations.NeptuneCallback):
        trainer.callback_handler.remove_callback(cb)

不过，如果变形金刚( Transformers )或海王星( Neptune )团队在这次回调中提供更多的灵活性，那就更好了。