diff options
| author | gdamms <damguillotin@gmail.com> | 2026-07-20 17:13:47 +0200 |
|---|---|---|
| committer | gdamms <damguillotin@gmail.com> | 2026-07-20 17:13:47 +0200 |
| commit | 8014df5feba6d44f5446e431a3285357224b8379 (patch) | |
| tree | 5a00fa9ab58fdad9ed6938c23983cc670448228a | |
| parent | f1fee867bbbffdd2686cc2f92313496f77d08b2f (diff) | |
| download | diffusion-mnist-main.tar.gz diffusion-mnist-main.zip | |
| -rw-r--r-- | README.md | 24 |
1 files changed, 0 insertions, 24 deletions
@@ -2,30 +2,6 @@ A diffusion-based generative model for MNIST digits implemented in PyTorch. -## Project Structure - -``` -diffusion-mnist/ -├── main.py # Main entry point with CLI -├── models/ # Neural network architectures -│ ├── __init__.py -│ ├── unet.py # UNet for diffusion model -│ └── autoencoder.py # Autoencoder for latent diffusion -├── src/ # Source code modules -│ ├── __init__.py -│ ├── config.py # Configuration and hyperparameters -│ ├── diffusion.py # Diffusion process utilities -│ ├── dataloader.py # Dataset and dataloader classes -│ ├── utils.py # Helper functions and metrics -│ ├── train_diffusion.py # Diffusion training script -│ ├── train_autoencoder.py # Autoencoder training script -│ └── sample.py # Sampling and visualization -├── checkpoints/ # Model checkpoints -├── plots/ # Generated visualizations -├── data/ # Dataset directory -└── runs/ # TensorBoard logs -``` - ## Installation ```bash |