Skip to content
Snippets Groups Projects
Commit f1cb653e authored by Egli Adrian (IT-SCI-API-PFI)'s avatar Egli Adrian (IT-SCI-API-PFI)
Browse files

small fix in object

parent a854eed7
No related branches found
No related tags found
No related merge requests found
Hide whitespace changes
Inline Side-by-side
reinforcement_learning/ppo/ppo_agent.py
+ 1
− 1
View file @ f1cb653e
...@@ -43,7 +43,7 @@ class DataBuffers: ...@@ -43,7 +43,7 @@ class DataBuffers:
class PPOModelNetwork(nn.Module): class PPOModelNetwork(nn.Module):
def __init__(self, state_size, action_size, hidsize1=128, hidsize2=128): def __init__(self, state_size, action_size, hidsize1=128, hidsize2=128):
super(DeepPPONetwork, self).__init__() super(PPOModelNetwork, self).__init__()
self.fc_layer_1_val = nn.Linear(state_size, hidsize1) self.fc_layer_1_val = nn.Linear(state_size, hidsize1)
self.shared_network = nn.Linear(hidsize1, hidsize2) self.shared_network = nn.Linear(hidsize1, hidsize2)
self.fc_policy_pi = nn.Linear(hidsize2, action_size) self.fc_policy_pi = nn.Linear(hidsize2, action_size)
......
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment