mirror of
https://github.com/huggingface/candle.git
synced 2025-06-15 02:16:37 +00:00
Do not implement Module for BatchNorm. (#1513)
This commit is contained in:
@ -147,7 +147,7 @@ fn conv(vb: VarBuilder, index: usize, p: usize, b: &Block) -> Result<(usize, Bl)
|
||||
let func = candle_nn::func(move |xs| {
|
||||
let xs = conv.forward(xs)?;
|
||||
let xs = match &bn {
|
||||
Some(bn) => bn.forward(&xs)?,
|
||||
Some(bn) => xs.apply_t(bn, false)?,
|
||||
None => xs,
|
||||
};
|
||||
let xs = if leaky {
|
||||
|
Reference in New Issue
Block a user