Skip to content

Commit

Permalink
Add parameter description
Browse files Browse the repository at this point in the history
  • Loading branch information
mieshkiwrk committed Oct 24, 2024
1 parent e26132a commit 925c376
Showing 1 changed file with 3 additions and 0 deletions.
3 changes: 3 additions & 0 deletions src/lightning/pytorch/trainer/trainer.py
Original file line number Diff line number Diff line change
Expand Up @@ -290,6 +290,9 @@ def __init__(
Default: ``os.getcwd()``.
Can be remote file paths such as `s3://mybucket/path` or 'hdfs://path/'
compile_fn: Provide torch.compile function to be applied after configuring strategy
Default: ``None``.
Raises:
TypeError:
If ``gradient_clip_val`` is not an int or float.
Expand Down

0 comments on commit 925c376

Please sign in to comment.