Skip to content

amithuli/order_matters_implementation

Repository files navigation

Order Matters: Sequence to Sequence for Sets (Vinyals et al., 2016)

Amit Huli & Sharon Mordechai

This project contains the implementation of the article: Order Matters: Sequence to Sequence for Sets Final Project Proposal (Vinyals et. al, 2016) The example given here is for the sorting experiment described in the article.

In order to run the experiments, use the experiments_runner.py (for the sorting experiment) or tsp_experiments_runner.py (for TSP experiment) and add the relevant hyperparameters accordingly for each experiment. Better to use it inside a notebook for better visibility.

Dependancies
  • PyTorch 1.12.1
  • TorchMetrics 0.9.3
  • NumPy 1.21.6
  • pytorch-lightning 1.7.4
  • tqdm 4.64.0

About

This repo contains an implementation of the paper:

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages