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

用法: update()

根据已安装的优化器和在前一个 forward-backward 批次中计算的梯度更新参数。

当使用 KVStore 更新参数进行多设备或多机训练时，参数的副本会存储在 KVStore 中。请注意，对于row_sparse 参数，此函数会更新 KVStore 中的参数副本，但不会将更新后的参数广播到所有设备/机器。请调用prepare广播row_sparse参数和下一批数据。

例子：

>>> # An example of updating module parameters.
>>> mod.init_optimizer(kvstore='local', optimizer='sgd',
...     optimizer_params=(('learning_rate', 0.01), ))
>>> mod.backward()
>>> mod.update()
>>> print mod.get_params()[0]['fc3_weight'].asnumpy()
[[  5.86930104e-03   5.28078526e-03  -8.88729654e-03  -1.08308345e-03
    6.13054074e-03   4.27560415e-03   1.53817423e-03   4.62131854e-03
    4.69872449e-03  -2.42400169e-03   9.94111411e-04   1.12386420e-03
    ...]]

相关用法

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