Skip to main content

​
TrainVisionModel

Defined in the train_vision module.

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