burn_train/logger/
in_memory.rs

1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
use super::Logger;

/// In memory logger.
#[derive(Default)]
pub struct InMemoryLogger {
    pub(crate) values: Vec<String>,
}

impl<T> Logger<T> for InMemoryLogger
where
    T: std::fmt::Display,
{
    fn log(&mut self, item: T) {
        self.values.push(item.to_string());
    }
}