diff options
Diffstat (limited to 'candle-examples/examples/reinforcement-learning/ddpg.rs')
| -rw-r--r-- | candle-examples/examples/reinforcement-learning/ddpg.rs | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/candle-examples/examples/reinforcement-learning/ddpg.rs b/candle-examples/examples/reinforcement-learning/ddpg.rs index 1ce4889e..5309eaf6 100644 --- a/candle-examples/examples/reinforcement-learning/ddpg.rs +++ b/candle-examples/examples/reinforcement-learning/ddpg.rs @@ -411,7 +411,7 @@ impl DDPG<'_> { pub fn actions(&mut self, state: &Tensor) -> Result<f32> { let actions = self .actor - .forward(&state.detach()?.unsqueeze(0)?)? + .forward(&state.detach().unsqueeze(0)?)? .squeeze(0)?; let actions = if self.train { (actions + self.ou_noise.sample()?)? |