Skip to main content

​
TrainVisionModel

Defined in the Train Vision module.
AI Agents Framework

​
Constructor

​
config_path
Any
default:"'config.yaml'"
No description available.

​
Methods

load_config()

Instance method.

print_system_info()

Instance method.

check_gpu()

Instance method.

check_ram()

Instance method.

prepare_model()

Instance method.

convert_sample()

Instance method.

load_datasets()

Instance method.

train_model()

Instance method.

vision_inference()

Instance method.

save_model_merged()

Instance method.

push_model_gguf()

Instance method.

save_model_gguf()

Instance method.

prepare_modelfile_content()

Instance method.

create_and_push_ollama_model()

Instance method.

run()

Instance method.

​
Source

View on GitHub

praisonai/train_vision.py at line 25

Agent Train

Training

Multimodal