Rooba
/
lightning
mirror of https://github.com/Lightning-AI/lightning.git
1
0
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Update trainer.py (#7340)

This commit is contained in:
ananthsub 2021-05-04 04:11:27 -07:00 committed by GitHub
parent 8c0ea92af2
commit 69cf63e2fd
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
pytorch_lightning/trainer/trainer.py
View File

@ -94,7 +94,7 @@ class Trainer(
checkpoint_callback: bool = True, checkpoint_callback: bool = True,
callbacks: Optional[Union[List[Callback], Callback]] = None, callbacks: Optional[Union[List[Callback], Callback]] = None,
default_root_dir: Optional[str] = None, default_root_dir: Optional[str] = None,
gradient_clip_val: float = 0, gradient_clip_val: float = 0.0,
gradient_clip_algorithm: str = 'norm', gradient_clip_algorithm: str = 'norm',
process_position: int = 0, process_position: int = 0,
num_nodes: int = 1, num_nodes: int = 1,