mistral_openapi_client/models/
hyperparameters.rs1use crate::models;
12use serde::{Deserialize, Serialize};
13
14#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
15pub struct Hyperparameters {
16 #[serde(rename = "training_steps", skip_serializing_if = "Option::is_none")]
17 pub training_steps: Option<i32>,
18 #[serde(rename = "learning_rate", skip_serializing_if = "Option::is_none")]
20 pub learning_rate: Option<f64>,
21 #[serde(rename = "weight_decay", skip_serializing_if = "Option::is_none")]
22 pub weight_decay: Option<f64>,
23 #[serde(rename = "warmup_fraction", skip_serializing_if = "Option::is_none")]
24 pub warmup_fraction: Option<f64>,
25 #[serde(rename = "epochs", skip_serializing_if = "Option::is_none")]
26 pub epochs: Option<f64>,
27 #[serde(rename = "seq_len", skip_serializing_if = "Option::is_none")]
28 pub seq_len: Option<i32>,
29 #[serde(rename = "fim_ratio", skip_serializing_if = "Option::is_none")]
30 pub fim_ratio: Option<f64>,
31}
32
33impl Hyperparameters {
34 pub fn new() -> Hyperparameters {
35 Hyperparameters {
36 training_steps: None,
37 learning_rate: None,
38 weight_decay: None,
39 warmup_fraction: None,
40 epochs: None,
41 seq_len: None,
42 fim_ratio: None,
43 }
44 }
45}
46