首页 > 解决方案 > 如何在pytorch中初始化nn.ConvTranspose2d的权重?

问题描述

如何nn.ConvTranspose2d通过双线性插值初始化权重?我发现nn.init女巫支持通过来自正态分布的随机数或零初始化来初始化每一层,但没有关于插值的内容。此外,插值可用于上采样,torch.nn.Upsample但通过这种方式,该层是不可训练的。关于如何使用双线性插值初始化可训练的 ConvTranspose2d 层的任何想法?

标签: pytorch

解决方案


推荐阅读