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

Python mxnet.module.BaseModule.init_optimizer用法及代碼示例


用法:

init_optimizer(kvstore='local', optimizer='sgd', optimizer_params=(('learning_rate', 0.01),), force_init=False)

參數：

  • kvstore：(str or KVStore) - 默認為‘local’.
  • optimizer：(str or Optimizer) - 默認為‘sgd’.
  • optimizer_params：(dict) - 默認為(('learning_rate', 0.01),).默認值不是字典，隻是為了避免 pylint 警告危險的默認值。
  • force_init：(bool) - 默認為False, 指示如果已安裝優化器，是否強製重新初始化它。
安裝和初始化優化器，以及初始化 kvstore

分布式訓練

例子：

>>> # An example of initializing optimizer.
>>> mod.init_optimizer(optimizer='sgd', optimizer_params=(('learning_rate', 0.005),))

相關用法


注：本文由純淨天空篩選整理自apache.org大神的英文原創作品 mxnet.module.BaseModule.init_optimizer。非經特殊聲明，原始代碼版權歸原作者所有，本譯文未經允許或授權，請勿轉載或複製。