(module, prefix='')
| 300 | state_dict._metadata = metadata |
| 301 | |
| 302 | def load(module, prefix=''): |
| 303 | local_metadata = {} if metadata is None else metadata.get( |
| 304 | prefix[:-1], {}) |
| 305 | module._load_from_state_dict( |
| 306 | state_dict, prefix, local_metadata, True, missing_keys, unexpected_keys, error_msgs) |
| 307 | for name, child in module._modules.items(): |
| 308 | if child is not None: |
| 309 | load(child, prefix + name + '.') |
| 310 | |
| 311 | load(model, prefix=prefix) |
| 312 |