7Gen - Advanced MNIST Digit Generation System
State-of-the-art Conditional GAN for MNIST digit synthesis with self-attention mechanisms.
π Features
- π― Conditional Generation: Generate specific digits (0β9) on demand.
- πΌοΈ High Quality Output: Sharp and realistic handwritten digit samples.
- β‘ Fast Inference: Real-time generation on GPU.
- π Easy Integration: Minimal setup, PyTorch-native implementation.
- π GPU Acceleration: Full CUDA support.
π Model Details
- Architecture: Conditional GAN with self-attention
- Parameters: 2.5M
- Input: 100-dimensional noise vector + class label
- Output: 28x28 grayscale images
- Training Data: MNIST dataset (60,000 images)
- Training Time: ~2 hours on NVIDIA RTX 3050 Ti
π§ͺ Performance Metrics
Metric | Score |
---|---|
FID Score | 12.3 |
Inception Score | 8.7 |
- Training Epochs: 100
- Batch Size: 64
βοΈ Training Configuration
model:
latent_dim: 100
num_classes: 10
generator_layers: [256, 512, 1024]
discriminator_layers: [512, 256]
training:
batch_size: 64
learning_rate: 0.0002
epochs: 100
optimizer: Adam
beta1: 0.5
beta2: 0.999
Inference Providers
NEW
This model isn't deployed by any Inference Provider.
π
Ask for provider support