Table Of Contents
Table Of Contents

KVStore

class mxnet.kvstore.KVStore(handle)[source]

A key-value store for synchronization of values, over multiple devices.

Init, push and pull

KVStore.init(key, value) Initializes a single or a sequence of key-value pairs into the store.
KVStore.pull(key[, out, priority, ignore_sparse]) Pulls a single value or a sequence of values from the store.
KVStore.push(key, value[, priority]) Pushes a single or a sequence of key-value pairs into the store.
KVStore.row_sparse_pull(key[, out, …]) Pulls a single RowSparseNDArray value or a sequence of RowSparseNDArray values from the store with specified row_ids.

Optimizer

KVStore.load_optimizer_states(fname) Loads the optimizer (updater) state from the file.
KVStore.save_optimizer_states(fname[, …]) Saves the optimizer (updater) state to a file.
KVStore.set_gradient_compression(…) Specifies type of low-bit quantization for gradient compression and additional arguments depending on the type of compression being used.
KVStore.set_optimizer(optimizer) Registers an optimizer with the kvstore.

Attributes

KVStore.num_workers Returns the number of worker nodes.
KVStore.rank Returns the rank of this worker node.
KVStore.type Returns the type of this kvstore.