Train Model • AI Agents Framework
TrainModel: Class reference for TrainModel
TrainModel
Defined in the train module.
AI Agents Framework
Constructor
ParamField
AnyNo description available.
Methods
load_config()
Instance method.
print_system_info()Instance method.
check_gpu()Instance method.
check_ram()Instance method.
prepare_model()Instance method.
process_dataset()Instance method.
tokenize_dataset()Instance method.
load_datasets()Instance method.
train_model()Instance method.
inference()Instance method.
load_model()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 GitHubpraisonai/train.py at line 108