conditional_diffusion_model
Folders and files
Name | Name | Last commit date | ||
---|---|---|---|---|
parent directory.. | ||||
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