flip.utils.model_weights_handling
=================================

.. py:module:: flip.utils.model_weights_handling


Functions
---------

.. autoapisummary::

   flip.utils.model_weights_handling.get_model_weights_diff


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

.. py:function:: get_model_weights_diff(original_weights: collections.OrderedDict, new_weights: collections.OrderedDict, iterations: int) -> nvflare.apis.dxo.DXO

   Compute the weights differences to send a weight update.

   :param original_weights: weights coming from the server (before training)
   :type original_weights: OrderedDict
   :param new_weights: weights coming out of the server (post training)
   :type new_weights: OrderedDict
   :param iterations: number of iterations
   :type iterations: int

   :returns: *DXO* -- DXO containing the weight updates for the server.