use crate::models;
use serde::{Deserialize, Serialize};
#[derive(Clone, Default, Debug, PartialEq, Serialize, Deserialize)]
pub struct Hyperparameters {
#[serde(rename = "training_steps", skip_serializing_if = "Option::is_none")]
pub training_steps: Option<i32>,
#[serde(rename = "learning_rate", skip_serializing_if = "Option::is_none")]
pub learning_rate: Option<f64>,
#[serde(rename = "weight_decay", skip_serializing_if = "Option::is_none")]
pub weight_decay: Option<f64>,
#[serde(rename = "warmup_fraction", skip_serializing_if = "Option::is_none")]
pub warmup_fraction: Option<f64>,
#[serde(rename = "epochs", skip_serializing_if = "Option::is_none")]
pub epochs: Option<f64>,
#[serde(rename = "seq_len", skip_serializing_if = "Option::is_none")]
pub seq_len: Option<i32>,
#[serde(rename = "fim_ratio", skip_serializing_if = "Option::is_none")]
pub fim_ratio: Option<f64>,
}
impl Hyperparameters {
pub fn new() -> Hyperparameters {
Hyperparameters {
training_steps: None,
learning_rate: None,
weight_decay: None,
warmup_fraction: None,
epochs: None,
seq_len: None,
fim_ratio: None,
}
}
}