Expose ItemLazy to be able to implement for custom types (#2525)

* Expose ItemLazy to be able to implement for custom types * Add comment
tracel-ai · Nov 22, 2024 · 9c31f75 · 9c31f75
1 parent 9a2b841
commit 9c31f75
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/crates/burn-train/src/metric/mod.rs b/crates/burn-train/src/metric/mod.rs
@@ -41,6 +41,9 @@ pub use memory_use::*;
 pub use top_k_acc::*;
 
 pub(crate) mod processor;
+// Expose `ItemLazy` so it can be implemented for custom types
+pub use processor::ItemLazy;
+
 /// Module responsible to save and exposes data collected during training.
 pub mod store;