search

AndyCao1125 / SDDPM

[WACV 2024] Spiking Denoising Diffusion Probabilistic Models
30 stars 3 forks source link

question for weight #6

Open chaibosong opened 3 months ago

chaibosong commented 3 months ago

Another small question, the cifa weight file you provided is about 1.9G, and the weight I trained myself is 763M. What is the reason for this?

AndyCao1125 commented 3 months ago

Thanks for your question. Because this file also holds the ema version of SNN-UNet (the same size as 'net_model') as well as other training objects such as optim, hence this ckpt file is large. You can check the items contained in ckpt by print("Keys in checkpoint:", list(ckpt.keys())).