python - TypeError: __init__() 得到了一个意外的关键字参数“文件路径”

Question

我不知道如何解决这个错误，但我希望你们中的一些人知道如何解决这个问题。

错误：TypeError: __init__() got an unexpected keyword argument 'filepath'

完整的错误信息：

  File "train.py", line 167, in <module>
    main(args)
  File "train.py", line 113, in main
    checkpoint_callback=checkpoint_callback(),
  File "train.py", line 86, in checkpoint_callback
    return ModelCheckpoint(
TypeError: __init__() got an unexpected keyword argument 'filepath'

from pytorch_lightning.callbacks import ModelCheckpoint

save_model_path = path/to/your/dir
def checkpoint_callback():
    return ModelCheckpoint(
        filepath= save_model_path,
        save_top_k=True,
        verbose=True,
        monitor='val_loss',
        mode='min',
        prefix=''
    )

Answer 1

ModelCheckpoint没有filepath关键字，但是它有一个dirpath关键字（如您在文档中看到的），替换filepath为dirpath，如下所示：

from pytorch_lightning.callbacks import ModelCheckpoint

save_model_path = path/to/your/dir
def checkpoint_callback():
    return ModelCheckpoint(
        dirpath=save_model_path, # changed line
        save_top_k=True,
        verbose=True,
        monitor='val_loss',
        mode='min',
        prefix=''
    )

Answer 2

pytorch_lightning 的版本过高或过低。api定义改变了

CLASS pytorch_lightning.callbacks.model_checkpoint.ModelCheckpoint(dirpath=None, filename=None,
monitor=None, verbose=False, save_last=None, save_top_k=1, save_weights_only=False,
mode='min', auto_insert_metric_name=True, every_n_train_steps=None, train_time_interval=None,
every_n_epochs=None, save_on_train_epoch_end=None, period=None, every_n_val_epochs=None)