python - TypeError：__init__() 得到意外的关键字参数“checkpoint_callback”

Question

我收到此错误消息：

TypeError                                 Traceback (most recent call last)
<ipython-input-41-2892cdd4e738> in <module>()
      5   max_epochs=N_EPOCHS,
      6   gpus=1, #GPU
----> 7   progress_bar_refresh_rate=30
      8 )

/usr/local/lib/python3.7/dist-packages/pytorch_lightning/utilities/argparse.py in insert_env_defaults(self, *args, **kwargs)
    343 
    344         # all args were already moved to kwargs
--> 345         return fn(self, **kwargs)
    346 
    347     return cast(_T, insert_env_defaults)

TypeError: __init__() got an unexpected keyword argument 'checkpoint_callback'

...当我运行这个块时：

trainer = pl.Trainer(
  logger=logger, 
  checkpoint_callback=checkpoint_callback,
  callbacks=[early_stopping_callback],
  max_epochs=N_EPOCHS,
  gpus=1, #GPU
  progress_bar_refresh_rate=30
)

“checkpoint_callback”定义如下：

checkpoint_callback = ModelCheckpoint(
  dirpath="checkpoints",
  filename="best-checkpoint",
  save_top_k=1,
  verbose=True,
  monitor="val_loss",
  mode="min"
)

我无法弄清楚导致错误的原因 - 谁能帮助我？

在此处查看完整源代码：https://colab.research.google.com/drive/1hT7PDVb0oGSpLejMGFBMWzRKTPwsSwwS?usp=sharing

Answer 1

当我查看 pytorch_lightning github 时，我在

init

中没有看到 checkpoint_callback 变量（https://github.com/Lightning-AI/lightning/blob/master/src/pytorch_lightning/trainer/trainer.py ）

你确定它就是这么叫的吗？通过这个

checkpoint_callback

你想达到什么目的？

//编辑： 我认为你只需将

checkpoint_callback

附加到

callbacks

列表

Answer 2

要保存最佳模型，您可以使用

enable_checkpointing

True 或 False。