Skip to main content

​
train_model

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

​
Signature

def train_model() -> Any

​
Returns

​
Returns
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

Agent Train

Training