fedlab_core.utils.serialization

Module Contents

Functions

ravel_model_params(model, cuda=False)	Squash model parameters or gradients into a single tensor.
unravel_model_params(model, parameter_update)	Assigns grad_update params to model.parameters.
unravel_model_grad(model, grad_update)	Assigns grad_update params to model.parameters.

fedlab_core.utils.serialization.ravel_model_params(model, cuda=False)

Squash model parameters or gradients into a single tensor.

fedlab_core.utils.serialization.unravel_model_params(model, parameter_update)

Assigns grad_update params to model.parameters. This is done by iterating through model.parameters() and assigning the relevant params in grad_update. NOTE: this function manipulates model.parameters.

fedlab_core.utils.serialization.unravel_model_grad(model, grad_update)

Assigns grad_update params to model.parameters. This is done by iterating through model.parameters() and assigning the relevant params in grad_update. NOTE: this function manipulates model.parameters().