sagemaker.train

sagemaker.train#

SageMaker Python SDK Train Module.

Modules

`agent_rft_job`	AgentRFTJob — wrapper around sagemaker-core Job for AgentRFT job category.
`aws_batch`
`base_trainer`
`common`
`configs`	DEPRECATED: This module has been moved to sagemaker.core.training.configs
`constants`	DEPRECATED: This module has been moved to sagemaker.core.training.constants
`container_drivers`	Sagemaker modules container drivers directory.
`custom_agent_lambda`	CustomAgentLambda — Lambda-based agent environment for Agentic RFT.
`defaults`	This module contains logic for setting defaults in ModelTrainer.
`distributed`	Distributed module.
`dpo_trainer`
`evaluate`	SageMaker Model Evaluation Module.
`local`
`model_trainer`	ModelTrainer class module.
`multi_turn_rl_trainer`	MultiTurnRLTrainer — trainer for Agentic Reinforcement Fine-Tuning (Multi-Turn RL) jobs.
`remote_function`	DEPRECATED: This module has been moved to sagemaker.core.remote_function
`rft`	SageMaker RFT SDK - Integration library for multi-turn RL training platform.
`rlaif_trainer`
`rlvr_trainer`
`sft_trainer`
`sm_recipes`
`templates`	Templates module.
`tuner`	Placeholder docstring
`types`	Types module.
`utils`	Utils module.