burn_train/logger/
in_memory.rs

1use super::Logger;
2
3/// In memory logger.
4#[derive(Default)]
5pub struct InMemoryLogger {
6    pub(crate) values: Vec<String>,
7}
8
9impl<T> Logger<T> for InMemoryLogger
10where
11    T: std::fmt::Display,
12{
13    fn log(&mut self, item: T) {
14        self.values.push(item.to_string());
15    }
16}