lightrft.strategy.deepspeed.deepspeed_utils
=================================================

.. currentmodule:: lightrft.strategy.deepspeed.deepspeed_utils

.. automodule:: lightrft.strategy.deepspeed.deepspeed_utils


get_train_ds_config
----------------------------------------------------------

.. autofunction:: get_train_ds_config



get_eval_ds_config
----------------------------------------------------------

.. autofunction:: get_eval_ds_config



offload_deepspeed_states
----------------------------------------------------------

.. autofunction:: offload_deepspeed_states



reload_deepspeed_states
----------------------------------------------------------

.. autofunction:: reload_deepspeed_states