huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-16 10:38:54 +00:00
Code Issues Packages Projects Releases Wiki Activity

Remove set_training. (#784)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-09 08:27:37 +01:00
committed by GitHub
parent 057f7909bc
commit c88d6fd4b9
1 changed files with 0 additions and 6 deletions
Download Patch File Download Diff File Expand all files Collapse all files

6
candle-core/src/lib.rs
View File

@ -112,12 +112,6 @@ impl ToUsize2 for (usize, usize) {
// A simple trait defining a module with forward method using a single argument. // A simple trait defining a module with forward method using a single argument.
pub trait Module: std::fmt::Debug { pub trait Module: std::fmt::Debug {
fn forward(&self, xs: &Tensor) -> Result<Tensor>; fn forward(&self, xs: &Tensor) -> Result<Tensor>;
/// 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 { impl Module for quantized::QMatMul {