Update trainer.py (#7340)

2021-05-04 04:11:27 -07:00 · 2021-05-04 04:11:27 -07:00 · 69cf63e2fd
parent 8c0ea92af2
commit 69cf63e2fd
1 changed files with 1 additions and 1 deletions
--- a/pytorch_lightning/trainer/trainer.py
+++ b/pytorch_lightning/trainer/trainer.py
@ -94,7 +94,7 @@ class Trainer(
        checkpoint_callback: bool = True,
        callbacks: Optional[Union[List[Callback], Callback]] = None,
        default_root_dir: Optional[str] = None,
-        gradient_clip_val: float = 0,
+        gradient_clip_val: float = 0.0,
        gradient_clip_algorithm: str = 'norm',
        process_position: int = 0,
        num_nodes: int = 1,