如何在变形金刚训练器运行中禁用海王星回调?

Question

安装完毕后。ai用于偶尔的ML实验日志记录，它被默认包含在所有变压器的回调列表中。教练运行。因此，它需要使用令牌进行正确的初始化，否则会抛出NeptuneMissingConfiguration错误，要求令牌和项目名称。这真的很烦人，我更喜欢海王星回调将自己限制为警告，或者如果没有提供令牌就禁用它。不幸的是，没有明显的方法来禁用这个回调，除非卸载海王星。人工智能。https://huggingface.co/docs/transformers/main_classes/callback上的文档页面说明这个回调默认是启用的，并且没有办法禁用它(不像其他一些可以通过环境变量禁用的回调)。

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

Answer 1

显然trainer初始化后的这段代码有帮助:

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

如果变形金刚或海王星团队在这个回调中提供更多的灵活性，那将是很好的。

Answer 2

要在变压器训练器运行中禁用海王星回调，可以将——no-neptune标志传递给trainer.train()函数。

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