PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS. Original implementation is available here.
- Pytorch
- torchvision
- numpy
- scipy
cd number_sorting
python train.py
cd jigsaw_puzzle
python train.py
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS. Original implementation is available here.
cd number_sorting
python train.py
cd jigsaw_puzzle
python train.py
PyTorch implementation of LEARNING LATENT PERMUTATIONS WITH GUMBEL-SINKHORN NETWORKS