From ac2fad59ac60b825fec68dec88cf60f2e40bbb9d Mon Sep 17 00:00:00 2001 From: Carlos Gutierrez Date: Sun, 16 Nov 2025 17:09:36 -0500 Subject: [PATCH] fixing memory --- config_quick_optimized.json | 14 +++++++------- 1 file changed, 7 insertions(+), 7 deletions(-) diff --git a/config_quick_optimized.json b/config_quick_optimized.json index ba54287..93ecd42 100644 --- a/config_quick_optimized.json +++ b/config_quick_optimized.json @@ -14,17 +14,17 @@ "use_optimized_attention": true }, "training": { - "batch_size": 16, - "max_epochs": 5, - "learning_rate": 2e-4, + "batch_size": 12, + "max_epochs": 50, + "learning_rate": 1e-4, "weight_decay": 0.01, - "warmup_steps": 500, + "warmup_steps": 1000, "max_grad_norm": 1.0, - "gradient_accumulation_steps": 8, + "gradient_accumulation_steps": 16, "use_amp": true, "save_dir": "./checkpoints_optimized", - "log_interval": 25, - "eval_interval": 200 + "log_interval": 50, + "eval_interval": 500 }, "data": { "data_dir": "./data",