# OFM

The supporting code for the paper "Optimal Flow Matching: Learning Straight Trajectories in Just One Step"

## Working with the repository.

The notebooks for Illustrative 2D *Gaussian*->*Eight Gaussians* experiments:

```bash
.\ofm_notebooks\toy8gau_*.ipynb
```

The script for Wasserstein-2 benchmark:

```bash
.\ofm_scripts\bench_script.py
```
Before running this scripts place the checkpoints from the github repository [https://github.com/iamalexkorotin/Wasserstein2Benchmark/tree/main/benchmarks/Mix3toMix10](https://github.com/iamalexkorotin/Wasserstein2Benchmark/tree/main/benchmarks/Mix3toMix10) in the folder `.\benchmarks\Mix3toMix10`.

The notebooks for ALAE experiment:

```bash
.\ofm_notebooks\alae_*.ipynb
```
