Praison AI

Train Model • AI Agents Framework

train_model: API reference for TrainVisionModel.train_model

train_model

Method

</div>

This is a method of the TrainVisionModel class in the train_vision module.

Signature

python
def train_model() -> Any

Returns

ResponseField Any

The result of the operation.

Uses

  • load_datasets
  • TrainingArguments
  • is_bf16_supported
  • os.getenv
  • SFTTrainer
  • UnslothVisionDataCollator
  • trainer.train
  • save_pretrained

Used By

Source

View on GitHub

praisonai/train_vision.py at line 150

Related Documentation

Agent Train Training