huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-19 11:56:45 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add clone to various nn layers. (#910)

Browse Source
This commit is contained in:
Laurent Mazare
2023-09-20 11:33:51 +01:00
committed by GitHub
parent f685b2231c
commit 7b1ddcff47
7 changed files with 11 additions and 11 deletions
Download Patch File Download Diff File Expand all files Collapse all files

2
candle-nn/src/batch_norm.rs
View File

@ -38,7 +38,7 @@ impl From<f64> for BatchNormConfig {
}
}
#[derive(Debug)]
#[derive(Clone, Debug)]
pub struct BatchNorm {
running_mean: Tensor,
running_var: Tensor,