in trainer/loss.py [0:0]
def cosine_sim(emb1, emb2): """compute cosine similarity of two embeddings Args: emb1 emb2 Returns: float: cosine similarity between (-1, 1) """ return emb1.mm(emb2.t())