Skip to content

Latest commit

 

History

History

conditional_diffusion_model

Class conditional diffusion model on NIST dataset, where classes are 'A','B','C','D','E','F','G','H' folders.

model weights can be found at veerav96/conditional-fingerprint-NIST Hugging Face

samples generated at intermeidate epochs during training at samples/ folder (each image has 8 sub images representing 8 classes)

Tensorboard event logs added

Training time : 4-5 hrs on V100 GPU