We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent e26132a commit 925c376Copy full SHA for 925c376
src/lightning/pytorch/trainer/trainer.py
@@ -290,6 +290,9 @@ def __init__(
290
Default: ``os.getcwd()``.
291
Can be remote file paths such as `s3://mybucket/path` or 'hdfs://path/'
292
293
+ compile_fn: Provide torch.compile function to be applied after configuring strategy
294
+ Default: ``None``.
295
+
296
Raises:
297
TypeError:
298
If ``gradient_clip_val`` is not an int or float.
0 commit comments