Praison AI

Train Model • AI Agents Framework

train_model: API reference for TrainModel.train_model

train_model

Method

</div>

This is a method of the TrainModel class in the train module.

Signature

python
def train_model() -> Any

Returns

ResponseField Any

The result of the operation.

Uses

  • load_datasets
  • tokenize_dataset
  • is_bfloat16_supported
  • os.getenv
  • TrainingArguments
  • SFTTrainer
  • train_on_responses_only
  • trainer.train
  • save_pretrained

Used By

Source

View on GitHub

praisonai/train.py at line 214

Related Documentation

Agent Train Training