diff options
| -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 |