search

songyingxin / Bert-TextClassification

Implemention some Baseline Model upon Bert for Text Classification
Other
677 stars 148 forks source link

BERTDPCNN.py有一个小问题 #10

Closed Clitost closed 4 years ago

Clitost commented 4 years ago

在forward的x = x.squeeze()这里，如果使用者最后一个batch的大小刚好是1时，会被squeeze掉，这样输出标签的维度不匹配，建议修改。