mirror of
https://github.com/wassname/DeepRL.git
synced 2026-10-07 11:13:51 +08:00
Rewrite state/reward normalizer
This commit is contained in:
1 parent
4c6481d5be
commit
8dfe6ff7c8
23 files changed
+125
-926
No files matched your search
@@ -90,7 +90,6 @@ class GaussianActorNet(nn.Module, BasicNet):
|
||||
action_scale=1,
|
||||
action_gate=F.tanh,
|
||||
gpu=-1,
|
||||
# unit_std=True,
|
||||
hidden_size=64,
|
||||
non_linear=F.tanh):
|
||||
super(GaussianActorNet, self).__init__()
|
||||
@@ -161,7 +160,3 @@ class DisjointActorCriticNet:
|
||||
def zero_grad(self):
|
||||
self.actor.zero_grad()
|
||||
self.critic.zero_grad()
|
||||
|
||||
def train(self):
|
||||
self.actor.train()
|
||||
self.critic.train()
|
||||
Reference in new issue
Block a user