along/HPCC2025
1
1
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

改网络的激活函数

Browse Source
This commit is contained in:
weixin_46229132 2025-03-14 09:22:40 +08:00
parent 3086413171
commit db890f83cf
1 changed files with 10 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
PPO2/PPO.py
View File

@ -48,17 +48,17 @@ class ActorCritic(nn.Module):
if has_continuous_action_space :
self.actor = nn.Sequential(
nn.Linear(state_dim, 64),
nn.Tanh(),
# nn.Tanh(),
# nn.Sigmoid(),
# nn.ReLU(),
nn.ReLU(),
nn.Linear(64, 64),
nn.Tanh(),
# nn.Tanh(),
# nn.Sigmoid(),
# nn.ReLU(),
nn.ReLU(),
nn.Linear(64, action_dim),
nn.Tanh()
# nn.Tanh()
# nn.Sigmoid()
# nn.ReLU()
nn.ReLU()
)
else:
self.actor = nn.Sequential(
@ -72,13 +72,13 @@ class ActorCritic(nn.Module):
# critic
self.critic = nn.Sequential(
nn.Linear(state_dim, 64),
nn.Tanh(),
# nn.Tanh(),
# nn.Sigmoid(),
# nn.ReLU(),
nn.ReLU(),
nn.Linear(64, 64),
nn.Tanh(),
# nn.Tanh(),
# nn.Sigmoid(),
# nn.ReLU(),
nn.ReLU(),
nn.Linear(64, 1)
)