This commit is contained in:
2025-11-21 02:57:50 -05:00
parent 8a7f9570c1
commit e6b5e43646
2 changed files with 15 additions and 1 deletions

View File

@@ -340,7 +340,7 @@ class Train(Train_Base):
def train(self, args):
# --------------------------------------- Learning parameters
n_envs = min(12, os.cpu_count())
n_envs = min(8, os.cpu_count())
n_steps_per_env = 1024 # RolloutBuffer is of size (n_steps_per_env * n_envs)
minibatch_size = 64 # should be a factor of (n_steps_per_env * n_envs)
total_steps = 50000000