Skip to content

PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS

Notifications You must be signed in to change notification settings

yanfang-research/gumbel-sinkhorn

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 

Repository files navigation

LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS

PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS. Original implementation is available here.

Requirement

  • Pytorch
  • torchvision
  • numpy
  • scipy

Number Sorting

cd number_sorting
python train.py

Jigsaw Puzzle

cd jigsaw_puzzle
python train.py

About

PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 100.0%