search

gitabtion / BertBasedCorrectionModels

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

这个模型可以在CPU上面运行吗？在train_SoftMaskedBert.yml文件里设置Device为CPU，把GPU_IDS注释掉之后还是会报错 #15

Closed A6Matrix closed 3 years ago

A6Matrix commented 3 years ago

image

image

gitabtion commented 3 years ago

可以，DEVICE那里改成小写的cpu即可