mirror of
https://github.com/wassname/pytorch-soft-actor-critic.git
synced 2026-06-27 18:06:10 +08:00
Clean Up
This commit is contained in:
@@ -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:
|
||||
|
||||
Reference in New Issue
Block a user