ruda-model 0.21.4

Ruda model configuration, module parameters, records and data loading.
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
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111

use alloc::string::String;

pub(super) fn record_type_name(name: &str) -> String {
    String::from(name)
}

#[cfg(test)]
mod tests {
    use super::record_type_name;
    use crate::record::{BinBytesRecorder, RudaRecord, FullPrecisionSettings};
    use core::any::type_name;

    #[test]
    fn preserves_builtin_recorder_metadata() {
        type R = BinBytesRecorder<FullPrecisionSettings>;
        let expected = "ruda_model::record::memory::BinBytesRecorder<ruda_model::record::settings::FullPrecisionSettings, alloc::vec::Vec<u8>>";
        assert_eq!(record_type_name(type_name::<R>()), expected);
        let record = RudaRecord::<_, crate::TestBackend>::new::<R>(());
        assert_eq!(record.metadata.format, expected);
        assert_eq!(record.metadata.version, "0.21.0");
    }

    #[test]
    fn preserves_custom_recorder_paths_and_maps_native_generic_arguments() {
        assert_eq!(
            record_type_name("client::Recorder<ruda_model::record::settings::FullPrecisionSettings>"),
            "client::Recorder<ruda_model::record::settings::FullPrecisionSettings>"
        );
        for name in [
            "client::Recorder<client::Settings>",
            "client::ruda_model::Recorder<client::ruda_model::Settings>",
            "my_ruda_model::Recorder",
            "client_ruda_model::record::memory::BinBytesRecorder",
        ] {
            assert_eq!(record_type_name(name), name);
        }
    }

    #[test]
    fn preserves_compound_type_structure() {
        assert_eq!(
            record_type_name("client::Recorder<(ruda_model::A, &ruda_model::B, [ruda_model::C; 2])>"),
            "client::Recorder<(ruda_model::A, &ruda_model::B, [ruda_model::C; 2])>"
        );
        assert_eq!(
            record_type_name("<ruda_model::A as client::Trait<ruda_model::B>>::Item"),
            "<ruda_model::A as client::Trait<ruda_model::B>>::Item"
        );
    }

    #[test]
    fn preserves_neural_network_generic_type_identities() {
        assert_eq!(
            record_type_name("client::Recorder<ruda_nn::modules::linear::Linear<client::Backend>, ruda_model::record::settings::FullPrecisionSettings>"),
            "client::Recorder<ruda_nn::modules::linear::Linear<client::Backend>, ruda_model::record::settings::FullPrecisionSettings>"
        );
        for name in [
            "client::ruda_nn::Recorder<client::ruda_nn::Settings>",
            "client_ruda_nn::Recorder",
            "ruda_unknown::Recorder",
        ] {
            assert_eq!(record_type_name(name), name);
        }
    }

    #[test]
    fn preserves_optimizer_generic_type_identities() {
        assert_eq!(
            record_type_name("client::Recorder<ruda_optim::optim::adam::AdamState<client::Backend, 2>, ruda_model::record::settings::FullPrecisionSettings>"),
            "client::Recorder<ruda_optim::optim::adam::AdamState<client::Backend, 2>, ruda_model::record::settings::FullPrecisionSettings>"
        );
        for name in [
            "client::ruda_optim::Recorder<client::ruda_optim::Settings>",
            "client_ruda_optim::Recorder",
        ] {
            assert_eq!(record_type_name(name), name);
        }
    }

    #[test]
    fn preserves_store_generic_type_identities() {
        assert_eq!(
            record_type_name("client::Recorder<ruda_store::adapter::HalfPrecisionAdapter>"),
            "client::Recorder<ruda_store::adapter::HalfPrecisionAdapter>"
        );
        for name in [
            "client::ruda_store::Recorder<client::ruda_store::Settings>",
            "client_ruda_store::Recorder",
        ] {
            assert_eq!(record_type_name(name), name);
        }
    }

    #[test]
    #[cfg(feature = "std")]
    fn preserves_file_recorder_and_precision_identities() {
        use crate::record::*;
        for (actual, expected) in [
            (type_name::<BinFileRecorder<FullPrecisionSettings>>(), "ruda_model::record::file::BinFileRecorder<ruda_model::record::settings::FullPrecisionSettings>"),
            (type_name::<BinGzFileRecorder<HalfPrecisionSettings>>(), "ruda_model::record::file::BinGzFileRecorder<ruda_model::record::settings::HalfPrecisionSettings>"),
            (type_name::<JsonGzFileRecorder<DoublePrecisionSettings>>(), "ruda_model::record::file::JsonGzFileRecorder<ruda_model::record::settings::DoublePrecisionSettings>"),
            (type_name::<PrettyJsonFileRecorder<FullPrecisionSettings>>(), "ruda_model::record::file::PrettyJsonFileRecorder<ruda_model::record::settings::FullPrecisionSettings>"),
            (type_name::<NamedMpkGzFileRecorder<HalfPrecisionSettings>>(), "ruda_model::record::file::NamedMpkGzFileRecorder<ruda_model::record::settings::HalfPrecisionSettings>"),
            (type_name::<DefaultFileRecorder<DoublePrecisionSettings>>(), "ruda_model::record::file::NamedMpkFileRecorder<ruda_model::record::settings::DoublePrecisionSettings>"),
            (type_name::<NamedMpkBytesRecorder<FullPrecisionSettings>>(), "ruda_model::record::memory::NamedMpkBytesRecorder<ruda_model::record::settings::FullPrecisionSettings>"),
            (type_name::<NoStdInferenceRecorder>(), "ruda_model::record::memory::BinBytesRecorder<ruda_model::record::settings::FullPrecisionSettings, &[u8]>"),
        ] {
            assert_eq!(record_type_name(actual), expected);
        }
    }
}