chore: add missing config to doc
This commit is contained in:
@@ -483,6 +483,7 @@ output_dir: ./completed-model
|
|||||||
# setting to `auto` will enable torch compile when torch>=2.5.1
|
# setting to `auto` will enable torch compile when torch>=2.5.1
|
||||||
torch_compile: # Optional[Union[Literal["auto"], bool]]
|
torch_compile: # Optional[Union[Literal["auto"], bool]]
|
||||||
torch_compile_backend: # Optional[str]
|
torch_compile_backend: # Optional[str]
|
||||||
|
torch_compile_mode: # 'default' | 'reduce-overhead' | 'max-autotune'
|
||||||
|
|
||||||
# Training hyperparameters
|
# Training hyperparameters
|
||||||
|
|
||||||
|
|||||||
Reference in New Issue
Block a user