hecongqing opened this issue 2 years ago · 0 comments
cos_sim = model.sim(query.unsqueeze(1), doc.unsqueeze(0))