Struct PlanConfig 

pub struct PlanConfig {

    pub task: String,
    pub data_path: PathBuf,
    pub val_path: Option<PathBuf>,
    pub test_path: Option<PathBuf>,
    pub model_size: String,
    pub model_path: Option<PathBuf>,
    pub num_classes: usize,
    pub output_dir: PathBuf,
    pub strategy: String,
    pub budget: usize,
    pub scout: bool,
    pub max_epochs: usize,
    pub manual_lr: Option<f32>,
    pub manual_lora_rank: Option<usize>,
    pub manual_batch_size: Option<usize>,
    pub manual_lora_alpha: Option<f32>,
    pub manual_warmup: Option<f32>,
    pub manual_gradient_clip: Option<f32>,
    pub manual_lr_min_ratio: Option<f32>,
    pub manual_class_weights: Option<String>,
    pub manual_target_modules: Option<String>,
}

Input configuration for plan generation.

This is the user’s intent — what they want to train. The plan builder validates this against reality and produces a TrainingPlan.

Fields

task: String

Task type (currently only “classify”).

data_path: PathBuf

Path to training data (JSONL).

val_path: Option<PathBuf>

Optional validation data (JSONL). If absent, split from training data.

test_path: Option<PathBuf>

Optional test data (JSONL). Used for post-train eval.

model_size: String

Model size hint (e.g. “0.5B”, “9B”).

model_path: Option<PathBuf>

Path to model weights directory.

num_classes: usize

Number of output classes.

output_dir: PathBuf

Output directory for checkpoints.

strategy: String

HPO strategy: “tpe”, “grid”, “random”, or “manual”.

budget: usize

HPO budget (number of trials). Ignored if strategy is “manual”.

scout: bool

Scout mode (1 epoch per trial).

max_epochs: usize

Maximum epochs per trial.

manual_lr: Option<f32>

Manual hyperparameters (used when strategy is “manual”).

manual_lora_rank: Option<usize>

Manual LoRA rank.

manual_batch_size: Option<usize>

Manual batch size.

manual_lora_alpha: Option<f32>

Manual LoRA alpha.

manual_warmup: Option<f32>

Manual warmup fraction.

manual_gradient_clip: Option<f32>

Manual gradient clip norm.

manual_lr_min_ratio: Option<f32>

Manual LR min ratio.

manual_class_weights: Option<String>

Manual class weight strategy.

manual_target_modules: Option<String>

Manual target modules.

Trait Implementations

impl Clone for PlanConfig

fn clone(&self) -> PlanConfig

Returns a duplicate of the value.

fn clone_from(&mut self, source: &Self)

Performs copy-assignment from source.

impl Debug for PlanConfig

fn fmt(&self, f: &mut Formatter<'_>) -> Result

Formats the value using the given formatter.

impl<'de> Deserialize<'de> for PlanConfig

fn deserialize<__D>(__deserializer: __D) -> Result<Self, __D::Error>

where __D: Deserializer<'de>,

Deserialize this value from the given Serde deserializer.

impl Serialize for PlanConfig

fn serialize<__S>(&self, __serializer: __S) -> Result<__S::Ok, __S::Error>

where __S: Serializer,

Serialize this value into the given Serde serializer.