search

microsoft / metric-transfer.pytorch

Deep Metric Transfer for Label Propagation with Limited Annotated Data
MIT License
49 stars 19 forks source link

Supervised Metric Pretraining #2

Open chingisooinar opened 3 years ago

chingisooinar commented 3 years ago

Hello, I am writing from my first account. I would like to add an option for supervised metric pretraining. It is used with LinearAverage from LinearAverage.py since the output is expected to be passed to this class. It performs eq. (4) as mentioned in https://arxiv.org/pdf/1812.08781.pdf. Also, I found using the temperature parameter handy, as mentioned in https://arxiv.org/pdf/1805.01978.pdf. Thanks for your consideration.

ghost commented 3 years ago

CLA assistant check
All CLA requirements met.

chingisooinar commented 3 years ago

Thank you, I refactored it and included a sample code for imagenet dataset.

bl0 commented 3 years ago

Thanks a lot for your quick reply. Can you post the training and finetuning log of the sample code? I want to check the correctness and performance.