使用自定义权重在 pytorch 中初始化权重?

时间:2021-04-03 04:37:40

标签: python pytorch torch

我想用来自特定分布的参数样本初始化我的神经网络的权重,这在 nn.init 模块中还没有。比方说 Beta 版发行版。

我看到您可以使用那里描述的方法:Custom weight initialization in PyTorch

但是你需要使用 nn.init。从我指定的分布中初始化权重的优雅方法是什么

0 个答案:

没有答案