From c88d6fd4b9615307c7689ec3133d09f16a6b7d77 Mon Sep 17 00:00:00 2001 From: Laurent Mazare Date: Sat, 9 Sep 2023 08:27:37 +0100 Subject: [PATCH] Remove set_training. (#784) --- candle-core/src/lib.rs | 6 ------ 1 file changed, 6 deletions(-) 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 {