Train Vision Model • AI Agents Framework

TrainVisionModel: Class reference for TrainVisionModel

TrainVisionModel

Defined in the Train Vision module.

AI Agents Framework

Constructor

ParamField Any

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

Related Documentation

Agent Train Training Multimodal