Fashion MNIST unconditional Unet Model trained using DDPM

Model Hyperparams:

Model size: 51,834,625 params
3 stages: 128, 256, 512 channels
Linear Attention in 2nd and 3rd stages, Self Attention in Middle Stage
Optimizer: Adam
LR: 3e-4
Batch Size: 64
Grad Accumulation: 8 steps
Effectibe Batch Size: 512
Total steps: 5,000
Linear Beta Schedule: 1000 Steps

Downloads last month: -; Downloads are not tracked for this model. How to track

Inference Providers NEW

This model is not currently available via any of the supported third-party Inference Providers, and HF Inference API was unable to determine this model's library.