首页
学习
活动
专区
圈层
工具
发布
首页
学习
活动
专区
圈层
工具
MCP广场
社区首页 >问答首页 >python:__init__()得到了一个意外的关键字参数'checkpoint_callback‘

python:__init__()得到了一个意外的关键字参数'checkpoint_callback‘
EN

Stack Overflow用户
提问于 2022-08-09 10:22:42
回答 1查看 2.1K关注 0票数 0

我收到了一条错误消息:

代码语言:javascript
运行
复制
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'

..。当我运行这部分时:

代码语言:javascript
运行
复制
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”的定义如下:

代码语言:javascript
运行
复制
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

EN

回答 1

Stack Overflow用户

回答已采纳

发布于 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列表中即可

票数 3
EN
页面原文内容由Stack Overflow提供。腾讯云小微IT领域专用引擎提供翻译支持
原文链接:

https://stackoverflow.com/questions/73290224

复制
相关文章

相似问题

领券
问题归档专栏文章快讯文章归档关键词归档开发者手册归档开发者手册 Section 归档