From dac3eee5f9867409b11cd853cd318ee9333f9336 Mon Sep 17 00:00:00 2001
From: "Egli Adrian (IT-SCI-API-PFI)" <adrian.egli@sbb.ch>
Date: Fri, 8 Jan 2021 09:39:35 +0100
Subject: [PATCH] Tensorboard support added
---
reinforcement_learning/ppo_agent.py | 2 +-
1 file changed, 1 insertion(+), 1 deletion(-)
diff --git a/reinforcement_learning/ppo_agent.py b/reinforcement_learning/ppo_agent.py
index d467cfe..9a43aff 100644
--- a/reinforcement_learning/ppo_agent.py
+++ b/reinforcement_learning/ppo_agent.py
@@ -34,7 +34,7 @@ class EpisodeBuffers:
class ActorCriticModel(nn.Module):
- def __init__(self, state_size, action_size, device, hidsize1=128, hidsize2=128):
+ def __init__(self, state_size, action_size, device, hidsize1=512, hidsize2=256):
super(ActorCriticModel, self).__init__()
self.device = device
self.actor = nn.Sequential(
--
GitLab