mmengine.runner.get_state_dict¶
- mmengine.runner.get_state_dict(module, destination=None, prefix='', keep_vars=False)[源代码]¶
Returns a dictionary containing a whole state of the module.
Both parameters and persistent buffers (e.g. running averages) are included. Keys are corresponding parameter and buffer names. This method is modified from
torch.nn.Module.state_dict()
to recursively check parallel module in case that the model has a complicated structure, e.g., nn.Module(nn.Module(DDP)).- 参数:
- 返回:
A dictionary containing a whole state of the module.
- 返回类型: