无法使用kaiming_normal初始化权重

Question

我正在尝试使用nn.init.kaiming_normal_（some_weight）初始化权重。但是，出现以下错误。

[AttributeError：模块'torch.nn'没有属性'init_kaiming_normal _']

我使用的是pytorch版本1.2.0，打印（火炬。版本）

据我在网上看到的，当我使用pytorch> = 0.4.0时，可以使用此功能。还有其他解决方案吗？

Answer 1

这是简单的错字吗？请注意，您的错误消息中提到了init_kaiming_normal_（在初始化后加下划线），而它应该是init.kaiming_normal_（在初始化后加点）。