Learning representation learning one step at a time.

First, clone the repository:

git clone https://github.com/Arktetra/relea.git

Then, install the project using:

pip install -e .

The training script can be run with a config as follows:

python train.py --config /configs/{model}/{config}.yaml

Example:

python train.py --config /configs/vae/vae.yaml

Then, the samples generated at certain epochs during the training can be viewed in work_dirs directory in the project root directory.

The following are some example notebooks:

• Flow Matching on a Toy Dataset