diff --git a/PPO.py b/PPO.py index 5383418..3b36d33 100644 --- 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),