search

gitabtion / BertBasedCorrectionModels

PyTorch impelementations of BERT-based Spelling Error Correction Models. 基于BERT的文本纠错模型，使用PyTorch实现。
Apache License 2.0
265 stars 43 forks source link

Fix: A faulty var name in `pl.Trainer`. #13

Closed okcd00 closed 3 years ago

okcd00 commented 3 years ago

This issue from the last commit makes the repo not work now.
The feeding vars for callbacks should be a list, the ckpt_callback is an instance of ModelCheckpoint.

image

okcd00 commented 3 years ago

With this fix commit, it works now.
image

okcd00 commented 3 years ago

Or maybe you can take callbacks=[ckpt_callback], which is also fine.

gitabtion commented 3 years ago

pytorch lightning的后续版本移除了ckpt_callback，所有的callback均使用callbacks传入，为兼容后续版本，烦请改为callbacks=[ckpt_callback]并再次pr谢谢。

okcd00 commented 3 years ago

@gitabtion Done