我收到了一条错误消息:
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
发布于 2022-08-09 10:42:22
当我查看pytorch_lightning github时,在init (https://github.com/Lightning-AI/lightning/blob/master/src/pytorch_lightning/trainer/trainer.py)中看不到checkpoint_callback
变量。
你确定这是怎么叫的吗?您想通过传递这个checkpoint_callback
来实现什么?
//编辑:我认为您只需将checkpoint_callback
添加到callbacks
列表中即可
https://stackoverflow.com/questions/73290224
复制相似问题