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