priba / graph_metric.pytorch

Graph Metric Learning in PyTorch

Graph Metric Learning

Graph Metric Learning in PyTorch.

Install

• Install all the requirements.

    $ conda env create -f environment.yml

Usage

    $ conda activate graphmetric

Train

• Write configuration file. Follow the example here.
• Run the training script with the corresponding configuration file ./train.sh config/train.cfg

Test

• Write configuration file. Follow the example here providing a load path (--load), and test option (-t).
• Run the test script with the corresponding configuration file ./train.sh config/test.cfg

Author

• Pau Riba (@priba)