Shortcuts

mmengine.model.update_init_info

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.