entrenar 0.7.10

Training & Optimization library with autograd, LoRA, quantization, and model merging
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78

//! Output Configuration
//!
//! Contains output and artifact configuration types for training manifests.

use serde::{Deserialize, Serialize};

/// Output and artifact configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct OutputConfig {
    /// Output directory (supports templates)
    pub dir: String,

    /// Model output configuration
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub model: Option<ModelOutputConfig>,

    /// Metrics export configuration
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub metrics: Option<MetricsOutputConfig>,

    /// Training report configuration
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub report: Option<ReportConfig>,

    /// Artifact registry configuration
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub registry: Option<RegistryConfig>,
}

/// Model output configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ModelOutputConfig {
    /// Output format (safetensors, pt, gguf, apr)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub format: Option<String>,

    /// Save optimizer state
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub save_optimizer: Option<bool>,

    /// Save scheduler state
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub save_scheduler: Option<bool>,
}

/// Metrics output configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct MetricsOutputConfig {
    /// Output format (parquet, csv, json)
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub format: Option<String>,

    /// Metrics to include
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub include: Option<Vec<String>>,
}

/// Training report configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct ReportConfig {
    pub enabled: bool,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub format: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub include_plots: Option<bool>,
}

/// Artifact registry configuration
#[derive(Debug, Clone, Serialize, Deserialize)]
pub struct RegistryConfig {
    pub enabled: bool,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub target: Option<String>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub include_config: Option<bool>,
    #[serde(default, skip_serializing_if = "Option::is_none")]
    pub include_metrics: Option<bool>,
}