File size: 778 Bytes
50ff975 3cc83f6 50ff975 3cc83f6 50ff975 3cc83f6 50ff975 3cc83f6 50ff975 3cc83f6 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 |
---
library_name: pytorch
tags:
- gan
- mnist
- image-generation
- computer-vision
- pytorch
license: mit
datasets:
- mnist
model-index:
- name: DCGAN-MNIST
results: []
---
# GAN-MNIST
A simple yet effective **DCGAN** trained on the **MNIST** dataset using PyTorch, designed for **data augmentation** experiments.
## 🧠 Model Details
- **Architecture:** Deep Convolutional GAN (DCGAN)
- **Generator/Discriminator:** 3-layer CNN
- **Latent Dimension:** 100
- **Epochs Trained:** 20
- **Final FID Score:** 24.16
- **Image Size:** 28×28 grayscale
## 📦 Usage
```python
from torch import load
from models.gan.model import Generator
# Load model
model = Generator(latent_dim=100, img_channels=1)
model.load_state_dict(load("generator.pt"))
model.eval()
|