diff --git a/candle-core/src/lib.rs b/candle-core/src/lib.rs index 3504b0a6..0a64a5a6 100644 --- a/candle-core/src/lib.rs +++ b/candle-core/src/lib.rs @@ -112,12 +112,6 @@ impl ToUsize2 for (usize, usize) { // A simple trait defining a module with forward method using a single argument. pub trait Module: std::fmt::Debug { fn forward(&self, xs: &Tensor) -> Result; - - /// Change the module to use training mode vs eval mode. - /// - /// The default implementation does nothing as this is only used for a couple modules such as - /// dropout or batch-normalization. - fn set_training(&mut self, _training: bool) {} } impl Module for quantized::QMatMul {