pytorch 的torch.nn模块中有几个BN类: nn.BatchNorm1d,nn.BatchNorm2d,nn.BatchNorm3d 。 主要参数有:. num_features :特征数. eps=1e-05 :ϵ,防止分母为0. momentum ...
確定! 回上一頁