Skip to main content

​
train_model

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

​
Signature

def train_model() -> Any

​
Returns

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

Agent Train

Training