Neural Optimal Transport

Project for the course "Generative Modeling for Images" of the MVA master at ENS Paris-Saclay.

Paper : Neural Optimal Transport, Alexander Korotin, Daniil Selikhanovych, Evgeny Burnaev, 2023.

This project uses the code from the official repository of the paper to reproduce the results on different datasets.

Report

Our report for this project is available here.

Code

• To train the model on the toy dataset, use this notebook.
• To train the model on the real dataset, use this script.

Contributors

@bastienlc, @paulbarbier