From 46553f0323a3299b56440ede739ddae7d7f9aab7 Mon Sep 17 00:00:00 2001 From: Vasilis Valatsos Date: Fri, 22 Mar 2024 13:41:52 +0100 Subject: [PATCH] Made agent dumber to speed up --- ml/ppo/brain.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ml/ppo/brain.py b/ml/ppo/brain.py index d6a403a..75153da 100644 --- a/ml/ppo/brain.py +++ b/ml/ppo/brain.py @@ -52,7 +52,7 @@ class PPOMemory: class ActorNetwork(nn.Module): - def __init__(self, input_dim, output_dim, alpha, fc1_dims=2048, fc2_dims=2048, chkpt_dir='tmp'): + def __init__(self, input_dim, output_dim, alpha, fc1_dims=1024, fc2_dims=1024, chkpt_dir='tmp'): super(ActorNetwork, self).__init__() self.chkpt_dir = chkpt_dir @@ -92,7 +92,7 @@ class ActorNetwork(nn.Module): class CriticNetwork(nn.Module): - def __init__(self, input_dims, alpha, fc1_dims=8192, fc2_dims=8192, chkpt_dir='tmp'): + def __init__(self, input_dims, alpha, fc1_dims=4096, fc2_dims=4096, chkpt_dir='tmp'): super(CriticNetwork, self).__init__() self.chkpt_dir = chkpt_dir