mode:‘fan_in’ (default)或者‘fan_out’. 使用fan_in保持weights的方差在前向传播中不变;使用fan_out保持weights的方差在反向传播中不变。
在使用过程中,一般都要默认设置fan_out
见Pytorch官网详解
欢迎分享,转载请注明来源:内存溢出
mode:‘fan_in’ (default)或者‘fan_out’. 使用fan_in保持weights的方差在前向传播中不变;使用fan_out保持weights的方差在反向传播中不变。
在使用过程中,一般都要默认设置fan_out
见Pytorch官网详解
欢迎分享,转载请注明来源:内存溢出
评论列表(0条)