handle torch_compile set to auto (#2172) [skip ci]

* handle torch_compile set to auto * update docs [skip ci] * add tests
2024-12-17 16:42:41 -05:00
parent 10cfecf02e
commit 3798229d85
4 changed files with 64 additions and 4 deletions
--- a/docs/config.qmd
+++ b/docs/config.qmd
@@ -337,7 +337,8 @@ comet_experiment_config: # Dictionary for additional configuration settings, see
 output_dir: ./completed-model

 # Whether to use torch.compile and which backend to use
-torch_compile:  # bool
+# setting to `auto` will enable torch compile when torch>=2.5.1
+torch_compile:  # Optional[Union[Literal["auto"], bool]]
 torch_compile_backend:  # Optional[str]

 # Training hyperparameters