From f2e5c358fa920e8aa1b9a486bc5fe691a6d24b90 Mon Sep 17 00:00:00 2001 From: mscs Date: Mon, 15 Apr 2024 22:22:30 -0400 Subject: [PATCH] =?UTF-8?q?OOM=20because=2064gb=20wasn't=20enough=20?= =?UTF-8?q?=F0=9F=98=B3?= MIME-Version: 1.0 Content-Type: text/plain; charset=UTF-8 Content-Transfer-Encoding: 8bit --- onefiveone/learn.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/onefiveone/learn.py b/onefiveone/learn.py index 8b777e72..b9a853f8 100644 --- a/onefiveone/learn.py +++ b/onefiveone/learn.py @@ -836,8 +836,8 @@ def train_model(env, total_steps, n_steps, batch_size, episode, file_name, save_ batch_size = 256 - n_steps = batch_size * 32 - total_steps = n_steps * 64 + n_steps = batch_size * 16 + total_steps = n_steps * 32 for e in range(0, episodes): model = train_model(env=run_env, total_steps=total_steps, n_steps = n_steps, batch_size = batch_size, episode=e,