sagemaker.train.common#

Classes

CustomizationTechnique(value)

Customization techniques for fine-tuning.

FineTuningOptions(options_dict)

Dynamic class for fine-tuning options with validation.

TrainingType(value)

Training types for fine-tuning.

class sagemaker.train.common.CustomizationTechnique(value)[source]#

Bases: Enum

Customization techniques for fine-tuning.

DPO = 'DPO'#
RLAIF = 'RLAIF'#
RLVR = 'RLVR'#
SFT = 'SFT'#
class sagemaker.train.common.FineTuningOptions(options_dict: Dict[str, Any])[source]#

Bases: object

Dynamic class for fine-tuning options with validation.

get_info(param_name: str = None)[source]#

Display parameter information in a user-friendly format.

to_dict() Dict[str, Any][source]#

Convert back to dictionary for hyperparameters with string values.

class sagemaker.train.common.TrainingType(value)[source]#

Bases: Enum

Training types for fine-tuning.

FULL = 'FULL'#
LORA = 'LORA'#