main.py contains all relevant code.

The following commands will reproduce the results in our paper:

python main.py --dataset Cars --loss hybrid --lr 0.00016 --root <folder_where_data_resides>
python main.py --dataset CUB --loss hybrid --lr 8e-05 --root <folder_where_data_resides>
python main.py --dataset SOP --loss hybrid --lr 4e-05 --root <folder_where_data_resides>
python main.py --dataset iNat --loss hybrid --lr 8e-05 --root <folder_where_data_resides>