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

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。非经特殊声明，原始代码版权归原作者所有，本译文未经允许或授权，请勿转载或复制。