From f1cb653e76328dd996f447587eadbd3871eb8e5a Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Thu, 3 Dec 2020 13:04:02 +0100
Subject: [PATCH] small fix in object
---
reinforcement_learning/ppo/ppo_agent.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/reinforcement_learning/ppo/ppo_agent.py b/reinforcement_learning/ppo/ppo_agent.py
index 07b2079..f213464 100644
--- a/reinforcement_learning/ppo/ppo_agent.py
+++ b/reinforcement_learning/ppo/ppo_agent.py
@@ -43,7 +43,7 @@ class DataBuffers:
class PPOModelNetwork(nn.Module):
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.shared_network = nn.Linear(hidsize1, hidsize2)
self.fc_policy_pi = nn.Linear(hidsize2, action_size)
--
GitLab