megengine.traced_module.TracedModule.replace_param

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

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

speedup multimachine training.

1.0 版后已移除.

上一页

megengine.traced_module.TracedModule.register_forward_pre_hook

下一页

megengine.traced_module.TracedModule.set_end_points