1 files changed, 2 insertions, 0 deletions
diff --git a/src/config.py b/src/config.py
index 25dcdda..4db66b0 100644
--- a/src/config.py
+++ b/src/config.py
@@ -39,6 +39,8 @@ class Config:
     log_n_examples: int = 10
     out_dir: str = "runs"
     device: str = "auto"             # "auto" | "cuda" | "cpu" | "mps"
+    use_amp: bool = True             # FP16 mixed precision on CUDA (Turing Tensor Cores)
+    compile_model: bool = False      # torch.compile (reduce-overhead / CUDA Graphs)
 
     @property
     def vocab(self) -> int: