diff options
| -rw-r--r-- | candle-nn/src/var_builder.rs | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/candle-nn/src/var_builder.rs b/candle-nn/src/var_builder.rs index 374260b0..311a6fb9 100644 --- a/candle-nn/src/var_builder.rs +++ b/candle-nn/src/var_builder.rs @@ -81,6 +81,10 @@ impl VarMap { tensor_data.insert(path.to_string(), var); Ok(tensor) } + + pub fn data(&self) -> &Mutex<HashMap<String, Var>> { + &self.data + } } // TODO: Maybe we would want the storage to be generic, e.g. with Box<dyn> to avoid too many |