【Consistency Models】复现指导以及效果分析。

【Consistency Models】复现指导以及效果分析。

  • 1、先看结果
  • 2、复现指导
  • 3、将结果可视化
  • 总结

1、先看结果

如果没看过论文的建议先看看总结笔记:【终结扩散模型】Consistency Models.OpenAI开源新模型代码,一步成图,1秒18张

用到的模型是cd_imagenet64_lpips.pt。
采样代码为:(依据显存来改batch_size的大小)

mpiexec -n 2 python image_sample.py --batch_size 64 --training_mode consistency_distillation 

你可能感兴趣的:(扩散模型+CV与NLP科研笔记,人脸相关科研笔记【AI换脸】,人工智能,计算机视觉,深度学习,Diffusion,Model,扩散模型)