mode:‘fan_in’ (default)
或者‘fan_out’.
使用fan_in
保持weights
的方差在前向传播中不变;使用fan_out
保持weights
的方差在反向传播中不变。
在使用过程中,一般都要默认设置fan_out
见Pytorch官网详解
Pytorch Kaiming 初始化(Initialization)中fan_in和fan_out的区别
最新推荐文章于 2025-04-22 09:02:26 发布
mode:‘fan_in’ (default)
或者‘fan_out’.
使用fan_in
保持weights
的方差在前向传播中不变;使用fan_out
保持weights
的方差在反向传播中不变。
在使用过程中,一般都要默认设置fan_out
见Pytorch官网详解