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 : if has_continuous_action_space :
self.actor = nn.Sequential( self.actor = nn.Sequential(
nn.Linear(state_dim, 64), nn.Linear(state_dim, 64),
nn.Tanh(), # nn.Tanh(),
# nn.Sigmoid(), # nn.Sigmoid(),
# nn.ReLU(), nn.ReLU(),
nn.Linear(64, 64), nn.Linear(64, 64),
nn.Tanh(), # nn.Tanh(),
# nn.Sigmoid(), # nn.Sigmoid(),
# nn.ReLU(), nn.ReLU(),
nn.Linear(64, action_dim), nn.Linear(64, action_dim),
nn.Tanh() # nn.Tanh()
# nn.Sigmoid() # nn.Sigmoid()
# nn.ReLU() nn.ReLU()
) )
else: else:
self.actor = nn.Sequential( self.actor = nn.Sequential(
@ -72,13 +72,13 @@ class ActorCritic(nn.Module):
# critic # critic
self.critic = nn.Sequential( self.critic = nn.Sequential(
nn.Linear(state_dim, 64), nn.Linear(state_dim, 64),
nn.Tanh(), # nn.Tanh(),
# nn.Sigmoid(), # nn.Sigmoid(),
# nn.ReLU(), nn.ReLU(),
nn.Linear(64, 64), nn.Linear(64, 64),
nn.Tanh(), # nn.Tanh(),
# nn.Sigmoid(), # nn.Sigmoid(),
# nn.ReLU(), nn.ReLU(),
nn.Linear(64, 1) nn.Linear(64, 1)
) )