diff options
| author | CaptainJack2491 <jayrupnakawala@gmail.com> | 2026-08-17 16:15:28 +0100 |
|---|---|---|
| committer | CaptainJack2491 <jayrupnakawala@gmail.com> | 2026-08-17 16:15:28 +0100 |
| commit | 66b8b77feb5fafd3696044ee33a80f26cb693a13 (patch) | |
| tree | 4058ae296735bca9d53bb5b9f2d742bc8b34fab0 /src/config.py | |
| parent | 535bb1dcf727e2273b956553c0491ec80ad8ec98 (diff) | |
feat(gpu): add CUDA support, chunked eval, bisect prime search, and scaling to 100k (48 tests green)
Diffstat (limited to 'src/config.py')
| -rw-r--r-- | src/config.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/src/config.py b/src/config.py index 60984db..25dcdda 100644 --- a/src/config.py +++ b/src/config.py @@ -34,9 +34,11 @@ class Config: early_stop_em: float = 1.0 early_stop_patience: int = 5 batch_size: int = 32 + eval_batch_size: int = 512 # chunked evaluation batch size to avoid GPU OOM max_out_len: int = 6 log_n_examples: int = 10 out_dir: str = "runs" + device: str = "auto" # "auto" | "cuda" | "cpu" | "mps" @property def vocab(self) -> int: |