【学习笔记8】阅读StyleID论文源码

论文【链接】
源码【链接】

一、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

参考

Pytorch之permute函数

三、parser.add_argument(action=‘store_true’)

action=‘store_true’表示：只要运行时该变量有传参就将该变量设为True。

参考

python之parser.add