diff options
| author | Laurent Mazare <laurent.mazare@gmail.com> | 2023-08-11 17:58:56 +0200 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2023-08-11 16:58:56 +0100 |
| commit | 55e428c8aed766d7e2433acd032f9d2768facf5f (patch) | |
| tree | 595e960cffa32854bd91d30cba96f2ff52d27853 | |
| parent | 01ea57da8c936e2b73b40848cf2eab9cadab94c9 (diff) | |
| download | candle-55e428c8aed766d7e2433acd032f9d2768facf5f.tar.gz candle-55e428c8aed766d7e2433acd032f9d2768facf5f.tar.bz2 candle-55e428c8aed766d7e2433acd032f9d2768facf5f.zip | |
Expose the varmap inner data. (#411)
| -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 |