在模型搭建中,需要用到不需要使用梯度下降更新的缓冲区值,比如扩散模型的方差,batch_normalize层的均值和方差。
读取方式为实例化模型类后,直接使用实例的该参数名称属性读取,比如object.mean就是读取类的实例object中缓冲区名为mean的值。
在模型搭建中,需要用到不需要使用梯度下降更新的缓冲区值,比如扩散模型的方差,batch_normalize层的均值和方差。
读取方式为实例化模型类后,直接使用实例的该参数名称属性读取,比如object.mean就是读取类的实例object中缓冲区名为mean的值。