mofushaohua/DML-Ensemble-of-Losses

DML-Ensemble-of-Losses

Pytorch Implementation of the paper "Ensemble of Loss Functions to Improve Generalizability of Deep Metric Learning methods"

To run the code:

Please  

1 - Upload one of the datasets and then specify its path in config.json file  

2 - Run demo_WEDL_run_best.py  

The code has been tested in following environment:  

Python 3.8.8  

Pytorch 1.8.1  

Good Luck