chore: add missing config to doc

2025-05-14 16:56:34 +07:00
parent 8a336a2c33
commit 930472b7c7
1 changed files with 1 additions and 0 deletions
--- a/docs/config.qmd
+++ b/docs/config.qmd
@@ -483,6 +483,7 @@ output_dir: ./completed-model
 # setting to `auto` will enable torch compile when torch>=2.5.1
 torch_compile:  # Optional[Union[Literal["auto"], bool]]
 torch_compile_backend:  # Optional[str]
+torch_compile_mode:  # 'default' | 'reduce-overhead' | 'max-autotune'

 # Training hyperparameters