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