search

JegZheng / truncated-diffusion-probabilistic-models

Pytorch implementation of TDPM
MIT License
31 stars 3 forks source link

May I ask how to test my trained model? #1

Open ImneCurline opened 1 year ago

ImneCurline commented 1 year ago

I didn' t find any test file, only denoinsing.py, however I don't know the spesific procedure to get the same result as your work

JegZheng commented 1 year ago

Thanks for your interest. To start the evaluation process, you may run main.py and add --sample or --test in the arguments and evaluate the FID with existing FID calculation libraries. I will update the repo with an automatic evaluation script. Stay tuned!