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

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

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


ModelOptimPair
----------------------------------------------------------

.. autodata:: ModelOptimPair



ModelOrModelOptimPair
----------------------------------------------------------

.. autodata:: ModelOrModelOptimPair



DeepspeedStrategy
----------------------------------------------------------

.. autoclass:: DeepspeedStrategy
    :members: __init__,create_optimizer,backward,optimizer_step,unwrap_model,prepare,prepare_reward_models,get_ds_train_config,get_ds_eval_config,init_model_context,moving_average,load_model,save_model,save_ckpt,load_ckpt