一、DDIM eta
ddim_step
表示执行几轮去噪迭代,eta
表示DDPM和DDIM的插值系数。当eta=0
时,为DDPM;当eta≠0
时,为DDIM。
参考
DDIM 简明讲解与 PyTorch 实现:加速扩散模型采样的通用方法
【stable diffusion】两个与采样器有关的参数效果:eta大小与ddim插值方式
二、torch.permute()函数
作用:
用于对torch进行维度变换。
transpose与permute的异同:
同:都是对tensor维度进行转置
异:permute函数可以对任意高维矩阵进行转置,但没有torch.permute()这个调用方式
对比:permute,也可以多次使用transpose
# 例子
torch.randn(2,3,4,5).permute(3,2,0,1).shape
参考
三、parser.add_argument(action=‘store_true’)
action=‘store_true’
表示:只要运行时该变量有传参就将该变量设为True。