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

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