Skip to content

Latest commit

 

History

History
14 lines (8 loc) · 598 Bytes

README.md

File metadata and controls

14 lines (8 loc) · 598 Bytes
Raw

Learning Distance Estimators from Pivoted Embeddings of Metric Objects

Code to reproduce the results presented in 'Learning Distance Estimators from Pivoted Embeddings of Metric Objects'.

How to reproduce

  1. Get and prepare the data (instructions here)

  2. pip install -r requirements.txt

  3. python train_all.py yfcc100m-hybridfc6

The train_all.py script adopts Ray Tune to handle and parallelize runs. You should modify the script and adjust the parameters in ray.init() and tune.run() to your resources and desired parallelism.