diff options
Diffstat (limited to 'candle-examples/examples/reinforcement-learning/policy_gradient.rs')
| -rw-r--r-- | candle-examples/examples/reinforcement-learning/policy_gradient.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/reinforcement-learning/policy_gradient.rs b/candle-examples/examples/reinforcement-learning/policy_gradient.rs index 6c355fe6..3ae2617d 100644 --- a/candle-examples/examples/reinforcement-learning/policy_gradient.rs +++ b/candle-examples/examples/reinforcement-learning/policy_gradient.rs @@ -14,7 +14,7 @@ fn new_model( ) -> Result<(impl Module, VarMap)> { let input_size = input_shape.iter().product(); - let mut varmap = VarMap::new(); + let varmap = VarMap::new(); let var_builder = VarBuilder::from_varmap(&varmap, dtype, device); let model = seq() |