Update PPO.py

nikhilbarhate99 · Dec 13, 2019 · 85a65d8 · 85a65d8
1 parent 64376aa
commit 85a65d8
Showing 1 changed file with 1 addition and 2 deletions.
diff --git a/PPO.py b/PPO.py
@@ -23,8 +23,7 @@ def clear_memory(self):
 class ActorCritic(nn.Module):
     def __init__(self, state_dim, action_dim, n_latent_var):
         super(ActorCritic, self).__init__()
-        self.affine = nn.Linear(state_dim, n_latent_var)
-
+
         # actor
         self.action_layer = nn.Sequential(
                 nn.Linear(state_dim, n_latent_var),