ksuit.utils.model.model_utils
=============================

.. py:module:: ksuit.utils.model.model_utils


Functions
---------

.. autoapisummary::

   ksuit.utils.model.model_utils.copy_params
   ksuit.utils.model.model_utils.update_ema


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

.. py:function:: copy_params(source_model, target_model)

   Copy parameters and buffers from source_model to target_model.

   :param source_model: The model to copy parameters from.
   :param target_model: The model to copy parameters to.


.. py:function:: update_ema(source_model, target_model, target_factor, copy_buffers = False)

   Update the target model with an exponential moving average of the source model.

   :param source_model: The source model to copy parameters from.
   :param target_model: The target model to update.
   :param target_factor: The factor to use for the exponential moving average.
   :param copy_buffers: Whether to copy buffers as well. Defaults to False.