Skip to content

Commit 925c376

Browse files
committed
Add parameter description
1 parent e26132a commit 925c376

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

Diff for: src/lightning/pytorch/trainer/trainer.py

+3
Original file line numberDiff line numberDiff line change
@@ -290,6 +290,9 @@ def __init__(
290290
Default: ``os.getcwd()``.
291291
Can be remote file paths such as `s3://mybucket/path` or 'hdfs://path/'
292292
293+
compile_fn: Provide torch.compile function to be applied after configuring strategy
294+
Default: ``None``.
295+
293296
Raises:
294297
TypeError:
295298
If ``gradient_clip_val`` is not an int or float.

0 commit comments

Comments
 (0)