:mod:`fedlab_core.utils.serialization`
======================================

.. py:module:: fedlab_core.utils.serialization


Module Contents
---------------


Functions
~~~~~~~~~

.. autoapisummary::

   fedlab_core.utils.serialization.ravel_model_params
   fedlab_core.utils.serialization.unravel_model_params
   fedlab_core.utils.serialization.unravel_model_grad


.. function:: ravel_model_params(model, cuda=False)

   Squash model parameters or gradients into a single tensor.


.. function:: 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`.


.. function:: unravel_model_grad(model, grad_update)

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