Skip to content

Commit ffcd941

Browse files
committed
Merge branch 'feature/double_precision' of https://github.com/ethanwharris/pytorch-lightning into feature/double_precision
2 parents ffb34c0 + 61a3e89 commit ffcd941

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

pytorch_lightning/plugins/precision/double.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -78,8 +78,8 @@ def connect(
7878
lr_schedulers: Sequence[Any],
7979
) -> Tuple['Module', Sequence['Optimizer'], Sequence[Any]]:
8080
"""Converts the model to double precision and wraps the `training_step`, `validation_step`, `test_step`,
81-
`predict_step`, and `forward` methods to convert incoming floating point data to double. Does not alter `optimizers`
82-
or `lr_schedulers`."""
81+
`predict_step`, and `forward` methods to convert incoming floating point data to double. Does not alter
82+
`optimizers` or `lr_schedulers`."""
8383
model = model.to(dtype=torch.float64)
8484
if isinstance(model, LightningModule):
8585
self.patches.append(_DoublePrecisionPatch.patch(model, 'training_step'))

0 commit comments

Comments
 (0)