> ## Documentation Index
> Fetch the complete documentation index at: https://praison.ai/docs/llms.txt
> Use this file to discover all available pages before exploring further.

# Train Vision Model • AI Agents Framework

> TrainVisionModel: Class reference for TrainVisionModel

# TrainVisionModel

> Defined in the [**Train Vision**](../modules/train_vision) module.

<Badge color="purple">AI Agents Framework</Badge>

## Constructor

<ParamField query="config_path" type="Any" required={false} default="'config.yaml'">
  No description available.
</ParamField>

## Methods

<CardGroup cols={2}>
  <Card title="load_config()" icon="function" href="../functions/TrainVisionModel-load_config">
    Instance method.
  </Card>

  <Card title="print_system_info()" icon="function" href="../functions/TrainVisionModel-print_system_info">
    Instance method.
  </Card>

  <Card title="check_gpu()" icon="function" href="../functions/TrainVisionModel-check_gpu">
    Instance method.
  </Card>

  <Card title="check_ram()" icon="function" href="../functions/TrainVisionModel-check_ram">
    Instance method.
  </Card>

  <Card title="prepare_model()" icon="function" href="../functions/TrainVisionModel-prepare_model">
    Instance method.
  </Card>

  <Card title="convert_sample()" icon="function" href="../functions/TrainVisionModel-convert_sample">
    Instance method.
  </Card>

  <Card title="load_datasets()" icon="function" href="../functions/TrainVisionModel-load_datasets">
    Instance method.
  </Card>

  <Card title="train_model()" icon="function" href="../functions/TrainVisionModel-train_model">
    Instance method.
  </Card>

  <Card title="vision_inference()" icon="function" href="../functions/TrainVisionModel-vision_inference">
    Instance method.
  </Card>

  <Card title="save_model_merged()" icon="function" href="../functions/TrainVisionModel-save_model_merged">
    Instance method.
  </Card>

  <Card title="push_model_gguf()" icon="function" href="../functions/TrainVisionModel-push_model_gguf">
    Instance method.
  </Card>

  <Card title="save_model_gguf()" icon="function" href="../functions/TrainVisionModel-save_model_gguf">
    Instance method.
  </Card>

  <Card title="prepare_modelfile_content()" icon="function" href="../functions/TrainVisionModel-prepare_modelfile_content">
    Instance method.
  </Card>

  <Card title="create_and_push_ollama_model()" icon="function" href="../functions/TrainVisionModel-create_and_push_ollama_model">
    Instance method.
  </Card>

  <Card title="run()" icon="function" href="../functions/TrainVisionModel-run">
    Instance method.
  </Card>
</CardGroup>

## Source

<Card title="View on GitHub" icon="github" href="https://github.com/MervinPraison/PraisonAI/blob/main/src/praisonai/praisonai/train_vision.py#L25">
  `praisonai/train_vision.py` at line 25
</Card>

***

## Related Documentation

<CardGroup cols={2}>
  <Card title="Agent Train" icon="dumbbell" href="/docs/concepts/agent-train" />

  <Card title="Training" icon="chart-line" href="/docs/train" />

  <Card title="Multimodal" icon="eye" href="/docs/features/multimodal" />
</CardGroup>