diff --git a/ddpg.py b/ddpg.py index e867218..8681e45 100644 --- a/ddpg.py +++ b/ddpg.py @@ -97,7 +97,7 @@ def select_action(self, state): mu = self.eval_anet(state) dist = Normal(mu, torch.tensor(self.var, dtype=torch.float)) action = dist.sample() - action.clamp(-2.0, 2.0) + action = action.clamp(-2.0, 2.0) return (action.item(),) def save_param(self):