Expand description
Declarative YAML configuration
This module provides Ludwig-style declarative training configuration via YAML.
§Example
model:
path: base-model.gguf
layers: [q_proj, v_proj]
data:
train: train.parquet
batch_size: 8
optimizer:
name: adam
lr: 1e-4
lora:
rank: 64
alpha: 16Structs§
- Architecture
Overrides - Architecture override parameters from YAML manifest.
- Audit
Args - Arguments for audit command
- Bench
Args - Arguments for bench command
- Bundle
Args - Arguments for bundle command
- Cite
Args - Arguments for cite command
- Cli
- Entrenar: Training & Optimization Library
- Column
Stats - Statistics about a column used for type inference
- Completion
Args - Arguments for completion command
- Curriculum
Stage - A curriculum learning stage: a data source active until a given step.
- Data
Config - Data configuration
- Deposit
Args - Arguments for deposit command
- Experiments
Args - Arguments for the experiments command
- Export
Args - Arguments for export command
- Finetune
Args - Arguments for the finetune command (plan/apply classification training)
- Inference
Config - Configuration for type inference
- Inferred
Schema - Inferred schema for a dataset
- Info
Args - Arguments for the info command
- Init
Args - Arguments for the init command
- Inspect
Args - Arguments for inspect command
- LoRA
Spec - LoRA configuration
- Merge
Args - Arguments for the merge command
- Merge
Spec - Model merging configuration
- Model
Ref - Model reference and target layers
- Monitor
Args - Arguments for monitor command
- Optim
Spec - Optimizer specification
- Preregister
Args - Arguments for preregister command
- Publish
Args - Arguments for the publish command
- Publish
Spec - Auto-publish configuration for uploading to HuggingFace Hub after training.
- Quant
Spec - Quantization configuration
- Quantize
Args - Arguments for the quantize command
- Research
Args - Arguments for the research command
- Research
Init Args - Arguments for research init command
- Train
Args - Arguments for the train command
- Train
Spec - Complete training specification
- Training
Params - Training hyperparameters
- Validate
Args - Arguments for the validate command
- Verify
Args - Arguments for verify command
Enums§
- Archive
Provider Arg - Archive provider for CLI
- Artifact
Type Arg - Artifact type for CLI
- Audit
Type - Audit type
- Citation
Format - Citation format for CLI
- Command
- Available commands
- Experiments
Command - Experiment store subcommands
- Export
Format - Export format for CLI
- Feature
Type - Inferred feature type
- Finetune
Command - Finetune subcommands (forjar-style plan/apply)
- Init
Template - Init template type
- Inspect
Mode - Inspection mode
- License
Arg - License for CLI
- Merge
Method - Merge method
- Model
Mode - Model execution mode
- Output
Format - Output format for info command
- Quant
Method - Quantization method
- Research
Command - Research subcommands
- Shell
Type - Shell type for completions
- Training
Method - Training method for –method flag
- Training
Mode - Training loss mode
- Validation
Error - Validation error type
Functions§
- apply_
overrides - Apply command-line overrides to a TrainSpec
- build_
model - Build a model from configuration by loading from file
- build_
optimizer - Build optimizer from configuration
- collect_
stats_ from_ samples - Collect statistics from sample values (simplified in-memory analysis)
- infer_
schema - Infer schema from column statistics
- infer_
schema_ from_ path - Placeholder: Load stats from Parquet file Real implementation would use arrow-rs/parquet crate
- infer_
type - Infer feature type from column statistics
- is_
hf_ repo_ id - Check if a string looks like a HuggingFace repo ID.
- load_
config - Load training spec from YAML file (without running training)
- parse_
args - Parse CLI arguments from a string slice (for testing)
- train_
from_ yaml - Train a model from YAML configuration file
- try_
load_ apr_ for_ inference - ALB-096: Try to load a model from APR format.
- validate_
config - Validate a training specification