We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
In the doc of dense_scores, it is mentioned that: torch.mean is dedicated to training
dense_scores
torch.mean is dedicated to training
neural-cherche/neural_cherche/utils/dense_scores.py
Lines 138 to 140 in 2df0214
However, in your training function of SparseEmb, you use torch.sum instead. So which one is preferable for training? I guess it should be torch.mean?
torch.sum
torch.mean
neural-cherche/neural_cherche/train/train_sparse_embed.py
Line 129 in 2df0214
The text was updated successfully, but these errors were encountered:
No branches or pull requests
In the doc of
dense_scores
, it is mentioned that:torch.mean is dedicated to training
neural-cherche/neural_cherche/utils/dense_scores.py
Lines 138 to 140 in 2df0214
However, in your training function of SparseEmb, you use
torch.sum
instead. So which one is preferable for training? I guess it should betorch.mean
?neural-cherche/neural_cherche/train/train_sparse_embed.py
Line 129 in 2df0214
The text was updated successfully, but these errors were encountered: