Merge pull request #17 from noanabeshima/patch-1

Minor change
nikhilbarhate99 · Dec 13, 2019 · 87eaa74 · 87eaa74
2 parents 64376aa + 85a65d8
commit 87eaa74
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),