Bert-Chinese-Text-Classification-Pytorch icon indicating copy to clipboard operation
Bert-Chinese-Text-Classification-Pytorch copied to clipboard

Published 20 hours ago • verified publisher icon 649453932

将输出类别个数等于label数

Open TheOnlylight opened this issue 3 years ago • 2 comments

将输出类别个数等于label数

Originally posted by @daonna in https://github.com/649453932/Bert-Chinese-Text-Classification-Pytorch/issues/17#issuecomment-569576271 我这边也是类似问题，但是就是主要在于class中放了22行，并且也是0-21数label，但是报错

TheOnlylight avatar Aug 06 '21 17:08 TheOnlylight

image 错误如图所示

TheOnlylight avatar Aug 06 '21 17:08 TheOnlylight

重启服务器或减小batch_size，自己就好了。。。

yanzengyun avatar Jun 19 '22 12:06 yanzengyun