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


Python mxnet.module.BaseModule.set_params用法及代码示例


用法:

set_params(arg_params, aux_params, allow_missing=False, force_init=True, allow_extra=False)

参数

  • arg_params(dict) - 名称到值的字典 (NDArray) 映射。
  • aux_params(dict) - 名称到值的字典 (NDArray) 映射。
  • allow_missing(bool) - 如果True, 参数可能包含缺失值,并且将调用初始化程序来填充这些缺失的参数。
  • force_init(bool) - 如果True, 即使已经初始化,也会强制重新初始化。
  • allow_extra(boolean, optional) - 是否允许符号不需要的额外参数。如果为 True,则当 arg_params 或 aux_params 包含执行程序不需要的额外参数时不会引发错误。

分配参数和辅助状态值。

例子

>>> # An example of setting module parameters.
>>> sym, arg_params, aux_params = mx.model.load_checkpoint(model_prefix, n_epoch_load)
>>> mod.set_params(arg_params=arg_params, aux_params=aux_params)

相关用法


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