mxnet.optimizer.updater

Updater class.

Functions

get_updater(optimizer)

Returns a closure of the updater needed for kvstore.

Classes

Updater(optimizer)

Updater for kvstore.
class mxnet.optimizer.updater.Updater(optimizer)[source]

Bases: object

Updater for kvstore.

get_states(dump_optimizer=False)[source]

Gets updater states.

Parameters:

dump_optimizer (bool, default False) – Whether to also save the optimizer itself. This would also save optimizer information such as learning rate and weight decay schedules.

set_states(states)[source]

Sets updater states.

sync_state_context(state, context)[source]

sync state context.

mxnet.optimizer.updater.get_updater(optimizer)[source]

Returns a closure of the updater needed for kvstore.

Parameters:

optimizer (Optimizer) – The optimizer.

Returns:

updater – The closure of the updater.

Return type:

function