From 7a1ab5882f73bac0890012e9c31e1b328c219f0c Mon Sep 17 00:00:00 2001 From: Priya Kasimbeg Date: Tue, 3 Feb 2026 07:06:13 +0000 Subject: [PATCH] add bsz for fwedu to reference algorithms for regression tests --- algorithms/archived_paper_baselines/adamw/jax/submission.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/algorithms/archived_paper_baselines/adamw/jax/submission.py b/algorithms/archived_paper_baselines/adamw/jax/submission.py index b8ea5d30a..c0ffe7601 100644 --- a/algorithms/archived_paper_baselines/adamw/jax/submission.py +++ b/algorithms/archived_paper_baselines/adamw/jax/submission.py @@ -254,6 +254,8 @@ def get_batch_size(workload_name): return 16 elif workload_name == 'cifar': return 32 + elif workload_name == 'finewebedu_lm': + return 64 else: raise ValueError(f'Unsupported workload name: {workload_name}.')