-
Notifications
You must be signed in to change notification settings - Fork 658
New issue
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
Set reasonable default values for all the losses and miners etc. #140
Comments
This would really help. Or at least give some examples with any values, starting from blank is the hardest. |
I'll try to include default values in the next version 0.9.90. Later I might include explanations + maybe images of equations from the papers. In the meantime, this table has "optimal" hyperparameters that I obtained through experiments on some loss functions: https://kevinmusgrave.github.io/powerful-benchmarker/papers/mlrc/#optimal-hyperparameters. (Side note: the negative These hyperparameter were obtained using bayesian optimization. The setup was:
The task was to finetune the pretrained model on 3 datasets, represented by each column: CUB (200 classes of bird images), Cars (196 classes of car images), and SOP (~20000 classes of images off of Ebay). |
Thank you for the quick reply and thank for the work on this wonderful library. BTW I'm trying to use ArcFace loss and it seems that your constructor doesn't pass the arguments required from the constructor of the base class https://github.com/KevinMusgrave/pytorch-metric-learning/blob/master/src/pytorch_metric_learning/losses/arcface_loss.py#L11 pytorch-metric-learning/src/pytorch_metric_learning/losses/large_margin_softmax_loss.py Line 15 in ad4b155
Unless I'm doing something wrong, the ArcFace paper is on my TODO list |
They should be getting passed via |
All the losses and miners have default values in v0.9.90, and equations have been added to many of the losses in the documentation. See the release notes to see breaking changes and new features. |
The parameters of some losses (e.g. ArcFace) and miners (e.g. MultiSimilarityMiner) don't have default init values, or suggested values in the documentation. This makes it harder for newcomers to use.
The text was updated successfully, but these errors were encountered: