search

zwkkk / wentian-rank2

“阿里灵杰”问天引擎电商搜索算法赛 第二名。电商领域两阶段文本匹配算法。
50 stars 18 forks source link

当验证集过大时，直接计算内存溢出 #4

Open hecongqing opened 1 year ago

hecongqing commented 1 year ago

cos_sim = model.sim(query.unsqueeze(1), doc.unsqueeze(0))