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 kNN evaluation code, the class probabilities retrieved from model output don't sum to 1:
dinov2/dinov2/eval/knn.py
Lines 192 to 195 in e1277af
This causes issues when computing kNN metrics. I would suggest adding the following line of code to ensure probabilities sum to 1:
def forward(self, features_dict, targets): for k in self.keys: features_dict = features_dict[k] features_dict = features_dict / features_dict.sum(dim=-1).unsqueeze(-1) return {"preds": features_dict, "target": targets}
The text was updated successfully, but these errors were encountered:
No branches or pull requests
In the kNN evaluation code, the class probabilities retrieved from model output don't sum to 1:
dinov2/dinov2/eval/knn.py
Lines 192 to 195 in e1277af
This causes issues when computing kNN metrics.
I would suggest adding the following line of code to ensure probabilities sum to 1:
The text was updated successfully, but these errors were encountered: