diff options
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: |