search

zoubohao / DenoisingDiffusionProbabilityModel-ddpm-

This may be the simplest implement of DDPM. You can directly run Main.py to train the UNet on CIFAR-10 dataset and see the amazing process of denoising.
MIT License
1.48k stars 156 forks source link

我首先运行完了Main.py,他训练了模型，然后我该怎么办，也不知道模型保存在哪，我想测试下下结果 #36

Open hxj2580 opened 7 months ago

hxj2580 commented 7 months ago

我首先运行完了Main.py,他训练了模型，然后我该怎么办，也不知道模型保存在哪，我想测试下下结果

keruoya commented 6 months ago

Main.py里state改成eval，模型在checkpoints里

zhangzhelala commented 6 months ago

好兄弟怎么解决的，我也是不知道怎么办了

keruoya commented 6 months ago

好兄弟怎么解决的，我也是不知道怎么办了

Main.py里state改成eval，模型在checkpoints里？