Gary-code / Sinkhorn-Network

Sinkhorn Network implementation with pytorch. Refer to paper in ICLR 2018: https://arxiv.org/pdf/1802.08665.pdf

Sinkhorn-Network

introduction blog

This is a simple Sinkhorn Network model implementation with Pytorch refer to paper. Here is the official tensorflow implementation.

Our Task

• Sorting the object permutations.

model file:

• object_class_glove.pkl
• genome-trainval.h5 (on my personal server)

dataset:

• folder: ./data/datasets/sentence_pairs.json

Data Preprocess

cd ./data
python prepor.py

After this step, train, validation and test data will generate under the ./data/datasets folder.

Main Process

Train and Validation

python train.py

Test

python test.py