huggingface/candle
1
0
Fork 0
mirror of https://github.com/huggingface/candle.git synced 2025-06-20 20:09:50 +00:00
Code Issues Packages Projects Releases Wiki Activity

Add get_ids to GradStore (#2379)

Browse Source
This commit is contained in:
Takanori MAEHARA
2024-08-01 09:56:13 +01:00
committed by GitHub
parent 1ba87a9450
commit ce90287f45
1 changed files with 5 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

5
candle-core/src/backprop.rs
View File

@ -756,4 +756,9 @@ impl GradStore {
}; };
Ok(grad) Ok(grad)
} }
/// Get the tensor ids of the stored gradient tensors
pub fn get_ids(&self) -> impl Iterator<Item = &TensorId> {
self.0.keys()
}
} }