# Diffusion Model Experiments
This directory contains the code for the diffusion model experiments for SD-KDE.
## Paths
Set the BASE_DIR environment variable to the path of the root directory of the project.
```
export BASE_DIR = "/path/to/minDiffusion"
```
## Dependencies
Pytorch and torchvision are required to run the code.
## Training
```
python train_2d.py
```
Trains a diffusion model on the 2D dataset.
