當前位置: 首頁>>代碼示例 >>用法及示例精選 >>正文


Python mxnet.gluon.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.SymbolBlock.imports。非經特殊聲明,原始代碼版權歸原作者所有,本譯文未經允許或授權,請勿轉載或複製。