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