1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22

pub use hypersearch_codegen::hyperparameters;

#[cfg(test)]
mod tests {
    use crate::hyperparameters;

    #[test]
    fn test_hyperparameter_search() {
        #[hyperparameters]
        struct Hyperparams {
            lr: Vec<f32>,
            embed_size: Vec<u16>,
        }

        let hyperparams = Hyperparams {
            lr: vec![1e-6, 1e-3],
            embed_size: vec![128, 1024],
        };

        assert_eq!(hyperparams.permutations().count(), 4);
    }
}