From 751cbe0d93af1fcc69147d27ca8babd63105e4dc Mon Sep 17 00:00:00 2001
From: Void Agent <void@jayrup.hermes>
Date: Sun, 16 Aug 2026 14:41:54 +0100
Subject: phase3: task_mode (is_prime), train_frac, lr_decay flags + tests (41
 green); Addendum 5 locked; 11-job batch

---
 src/config.py | 3 +++
 1 file changed, 3 insertions(+)

(limited to 'src/config.py')

diff --git a/src/config.py b/src/config.py
index 0b161e7..60984db 100644
--- a/src/config.py
+++ b/src/config.py
@@ -8,9 +8,11 @@ from dataclasses import dataclass, fields
 class Config:
     # data
     vocab_mode: str = "digits"       # "digits" | "integers"
+    task_mode: str = "next_prime"    # "next_prime" | "is_prime"
     range_start: int = 2
     range_end: int = 100             # inclusive
     holdout_frac: float = 0.30
+    train_frac: float = 1.0          # 1.0 = use full train split; 0.4/0.5 for E5
     seed: int = 0
     # model
     model: str = "rnn"               # "rnn" | "transformer"
@@ -25,6 +27,7 @@ class Config:
     n_heads: int = 4
     # training
     lr: float = 1e-3
+    lr_decay: bool = False           # cosine 1e-3 -> 1e-4 over the run (E3)
     weight_decay: float = 1.0
     max_train_steps: int = 200_000
     eval_every: int = 200
-- 
cgit v1.2.3