search

zhoujx4 / DuReader-Checklist-BASELINE

百度2021年语言与智能技术竞赛机器阅读理解torch版baseline
53 stars 16 forks source link

A Small Bug #3

Open W2Q3Q1 opened 3 years ago

W2Q3Q1 commented 3 years ago

models/model.py line23, 24需要给squeeze函数指定参数dim=-1，否则在设置batch_size不当的情况下，可能产生单样本batch，进而导致此函数产生的输出由应有的2维tensor变成1维tensor，引起崩溃。

zhoujx4 commented 3 years ago

你好，十分感谢你的两个建议，github已经更新。

------------------ 原始邮件 ------------------ 发件人: "zhoujx4/DuReader-Checklist-BASELINE" @.>; 发送时间: 2021年4月24日(星期六) 晚上9:55 @.>; @.***>; 主题: [zhoujx4/DuReader-Checklist-BASELINE] A Small Bug (#3)

models/model.py line23, 24需要给squeeze函数指定参数dim=-1，否则在设置batch_size不当的情况下，可能产生单样本batch，进而导致此函数产生的输出由应有的2维tensor变成1维tensor，引起崩溃。

— You are receiving this because you are subscribed to this thread. Reply to this email directly, view it on GitHub, or unsubscribe.