burn_core/lr_scheduler/
constant.rs

1use burn_tensor::backend::Backend;
2
3use super::LrScheduler;
4use crate::LearningRate;
5
6/// Constant learning rate implementing [learning rate scheduler](LrScheduler).
7///
8/// # Notes
9///
10/// You can also use [learning rate](LearningRate) which the same effect.
11#[derive(new, Clone, Debug)]
12pub struct ConstantLr {
13    lr: LearningRate,
14}
15
16impl From<LearningRate> for ConstantLr {
17    fn from(lr: LearningRate) -> Self {
18        Self { lr }
19    }
20}
21
22impl LrScheduler for ConstantLr {
23    type Record<B: Backend> = ();
24
25    fn step(&mut self) -> LearningRate {
26        self.lr
27    }
28
29    fn to_record<B: Backend>(&self) -> Self::Record<B> {}
30
31    fn load_record<B: Backend>(self, _record: Self::Record<B>) -> Self {
32        self
33    }
34}
35
36impl LrScheduler for LearningRate {
37    type Record<B: Backend> = ();
38
39    fn step(&mut self) -> LearningRate {
40        *self
41    }
42
43    fn to_record<B: Backend>(&self) -> Self::Record<B> {}
44
45    fn load_record<B: Backend>(self, _record: Self::Record<B>) -> Self {
46        self
47    }
48}
burn_core/lr_scheduler/constant.rs

burn_core/lr_scheduler/
constant.rs
