mindspore.load_checkpoint
- mindspore.load_checkpoint(ckpt_file_name, net=None, strict_load=False, filter_prefix=None, dec_key=None, dec_mode='AES-GCM', specify_prefix=None)[源代码]
加载checkpoint文件。
Note
specify_prefix 和 filter_prefix 的功能相互之间没有影响。
如果发现没有参数被成功加载,将会报ValueError.
参数:
ckpt_file_name (str) – checkpoint的文件名称。
net (Cell) – 加载checkpoint参数的网络。默认值:None。
strict_load (bool) – 是否将严格加载参数到网络中。如果是False, 它将根据相同的后缀名将参数字典中的参数加载到网络中,并会在精度不匹配时,进行强制精度转换,比如将 float32 转换为 float16 。默认值:False。
filter_prefix (Union[str, list[str], tuple[str]]) – 以 filter_prefix 开头的参数将不会被加载。默认值:None。
dec_key (Union[None, bytes]) – 用于解密的字节类型密钥,如果值为None,则不需要解密。默认值:None。
dec_mode (str) – 该参数仅当 dec_key 不为None时有效。指定解密模式,目前支持“AES-GCM”和“AES-CBC”。默认值:“AES-GCM”。
specify_prefix (Union[str, list[str], tuple[str]]) – 以 specify_prefix 开头的参数将会被加载。默认值:None。
返回:
字典,key是参数名称,value是Parameter类型。当使用
mindspore.save_checkpoint()
的 append_dict 参数和CheckpointConfig
的 append_info 参数保存 checkpoint, append_dict 和 append_info 是dict类型,且它们的值value是string时,加载checkpoint得到的返回值是string类型,其它情况返回值均是Parameter类型。异常:
ValueError – checkpoint文件格式不正确。
ValueError – 没有一个参数被成功加载。
TypeError specify_prefix 或者 filter_prefix 的数据类型不正确。
样例:
>>> import mindspore as ms >>> >>> ckpt_file_name = "./checkpoint/LeNet5-1_32.ckpt" >>> param_dict = ms.load_checkpoint(ckpt_file_name, filter_prefix="conv1", specify_prefix="conv", ) >>> print(param_dict["conv2.weight"]) Parameter (name=conv2.weight, shape=(16, 6, 5, 5), dtype=Float32, requires_grad=True)