mmengine.model.update_init_info — mmengine 0.10.7 documentation (original) (raw)

• Docs >
• mmengine.model >
• mmengine.model.update_init_info
• 以中文阅读

Shortcuts

mmengine.model.update_init_info(module, init_info)[source]¶

Update the _params_init_info in the module if the value of parameters are changed.

Parameters:

• ( obj (module) – nn.Module): The module of PyTorch with a user-defined attribute _params_init_info which records the initialization information.

• init_info (str) – The string that describes the initialization.

• mmengine.model.update_init_info

  • update_init_info()