当前位置: 首页>>代码示例 >>用法及示例精选 >>正文


Python mxnet.gluon.nn.SymbolBlock.imports用法及代码示例


用法:

static imports(symbol_file, input_names, param_file=None, ctx=None, allow_missing=False, ignore_extra=False)

参数

  • symbol_file(str) - 符号文件的路径。
  • input_names(list of str) - 输入变量名称列表
  • param_file(str, optional) - 参数文件的路径。
  • ctx(mxnet.context.Context, default None) - 要初始化的上下文gluon.SymbolBlock在。
  • allow_missing(bool, default False) - 是否静默跳过文件中未表示的加载参数。
  • ignore_extra(bool, default False) - 是否静默忽略此块中不存在的文件中的参数。

返回

gluon.SymbolBlock 从符号和参数文件中加载。

返回类型

mxnet.gluon.SymbolBlock

导入以前由gluon.HybridBlock.exportModule.save_checkpoint 保存的模型作为gluon.SymbolBlock 用于 Gluon。

例子

>>> net1 = gluon.model_zoo.vision.resnet18_v1(
...     prefix='resnet', pretrained=True)
>>> net1.hybridize()
>>> x = mx.nd.random.normal(shape=(1, 3, 32, 32))
>>> out1 = net1(x)
>>> net1.export('net1', epoch=1)
>>>
>>> net2 = gluon.SymbolBlock.imports(
...     'net1-symbol.json', ['data'], 'net1-0001.params')
>>> out2 = net2(x)

相关用法


注:本文由纯净天空筛选整理自apache.org大神的英文原创作品 mxnet.gluon.nn.SymbolBlock.imports。非经特殊声明,原始代码版权归原作者所有,本译文未经允许或授权,请勿转载或复制。