diff --git a/docs/config.qmd b/docs/config.qmd index f03c9da06..2e0e25987 100644 --- 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