---
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()