mindspore.load
- mindspore.load(file_name, **kwargs)[source]
Load MindIR.
The returned object can be executed by a GraphCell, see class
mindspore.nn.GraphCell
for more details.- Parameters
file_name (str) – MindIR file name.
kwargs (dict) –
Configuration options dictionary.
dec_key (bytes): Byte-type key used for decryption. The valid length is 16, 24, or 32.
dec_mode (Union[str, function]): Specifies the decryption mode, to take effect when dec_key is set.
Option: ‘AES-GCM’, ‘AES-CBC’, ‘SM4-CBC’ or customized decryption. Default:
'AES-GCM'
.For details of using the customized decryption, please check the tutorial.
obf_func (function): A python function used for loading obfuscated MindIR model, which can refer to obfuscate_model().
- Returns
GraphCell, a compiled graph that can executed by GraphCell.
- Raises
ValueError – MindIR file does not exist or file_name is not a string.
RuntimeError – Failed to parse MindIR file.
Examples
>>> import numpy as np >>> import mindspore as ms >>> import mindspore.nn as nn >>> from mindspore import Tensor >>> from mindspore import context >>> context.set_context(mode=context.GRAPH_MODE) >>> >>> net = nn.Conv2d(1, 1, kernel_size=3, weight_init="ones") >>> input_tensor = Tensor(np.ones([1, 1, 3, 3]).astype(np.float32)) >>> ms.export(net, input_tensor, file_name="net", file_format="MINDIR") >>> graph = ms.load("net.mindir") >>> net = nn.GraphCell(graph) >>> output = net(input_tensor) >>> print(output) [[[[4. 6. 4.] [6. 9. 6.] [4. 6. 4.]]]]
- Tutorial Examples: