Function load

modelscope/utils/checkpoint.py:264–272

(module: nn.Module, prefix='')

        else:

            def load(module: nn.Module, prefix=''):
                local_metadata = {} if metadata is None else metadata.get(
                    prefix[:-1], {})
                args = (state_dict, prefix, local_metadata, True, [], [],
                        error_msgs)
                module._load_from_state_dict(*args)
                for name, child in module._modules.items():
                    if child is not None:
                        load(child, prefix + name + '.')

            load(model, prefix=start_prefix)