diff --git a/sac.py b/sac.py index bc1ab2a..b118d0c 100644 --- a/sac.py +++ b/sac.py @@ -34,7 +34,6 @@ class SAC(object): self.policy = GaussianPolicy(num_inputs, action_space.shape[0], args.hidden_size).to(self.device) - self.policy = GaussianPolicy(self.num_inputs, self.action_space, args.hidden_size).to(self.device) self.policy_optim = Adam(self.policy.parameters(), lr=args.lr) else: