search

zhaogaofeng611 / TextMatch

基于Pytorch的，中文语义相似度匹配模型（ABCNN、Albert、Bert、BIMPM、DecomposableAttention、DistilBert、ESIM、RE2、Roberta、SiaGRU、XlNet）
787 stars 146 forks source link

推理代码有BUG #15

Open JasonCZH4 opened 8 months ago

JasonCZH4 commented 8 months ago

在test.py文件中，DataLoader(test_data, shuffle=True, batch_size=batch_size) shuffle应该设为False，这是一个很低级的BUG，卡了我一天，在最意想不到的地方有BUG