megengine.module.LayerNorm.replace_param

LayerNorm.replace_param(params, start_pos, seen=None)

Replaces module’s parameters with params, used by ParamPack to

speedup multimachine training.

1.0 版后已移除.

上一页

megengine.module.LayerNorm.register_forward_pre_hook

下一页

megengine.module.LayerNorm.reset_parameters