Update model.py

2026-08-13 12:30:39 +08:00 · 2018-12-05 02:28:37 +05:30
parent 9b7d14c5c6
commit 6440e47a0b
1 changed files with 1 additions and 1 deletions
@@ -121,7 +121,7 @@ class DeterministicPolicy(nn.Module):
        x = inputs
        x = F.relu(self.linear1(x))
        x = F.relu(self.linear2(x))
-        mean = F.tanh(self.mu(x))
+        mean = F.tanh(self.mean(x))
        return mean