search

SeanLee97 / AnglE

Train and Infer Powerful Sentence Embeddings with AnglE | 🔥 SOTA on STS and MTEB Leaderboard
https://arxiv.org/abs/2309.12871
MIT License
397 stars 30 forks source link

some mistake #52

Closed HaloKim closed 4 months ago

HaloKim commented 4 months ago

in angle_emb/angle.py

line 767

labels = inputs.pop("labels", None) <-- may be error

# labels = inputs.pop("labels", None) <-- may be ok

You have already declared.

def compute_loss(self, model, inputs, return_outputs=False):
        labels = inputs.pop("labels", None) <-- like this
SeanLee97 commented 4 months ago

@HaloKim thx! I’ve fixed this issue in v0.3.5