SageMaker Train
===============

Training capabilities including model training, hyperparameter tuning, and distributed training.

.. currentmodule:: sagemaker.train

Model Training
--------------

.. automodule:: sagemaker.train
   :members:
   :undoc-members:
   :show-inheritance:

Distributed Training
--------------------

.. automodule:: sagemaker.train.distributed
   :members:
   :undoc-members:
   :show-inheritance:

Multi-Turn RL Training
----------------------

.. automodule:: sagemaker.train.multi_turn_rl_trainer
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: sagemaker.train.agent_rft_job
   :members:
   :undoc-members:
   :show-inheritance:

.. automodule:: sagemaker.train.custom_agent_lambda
   :members:
   :undoc-members:
   :show-inheritance:

Model Evaluation
----------------

.. automodule:: sagemaker.train.evaluate
   :members:
   :undoc-members:
   :show-inheritance:

Multi-Turn RL Evaluation
------------------------

.. automodule:: sagemaker.train.evaluate.multi_turn_rl_evaluator
   :members:
   :undoc-members:
   :show-inheritance: