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 5f3ede3 commit c6cc85fCopy full SHA for c6cc85f
pytorch_lightning/core/decorators.py
@@ -71,7 +71,6 @@ def auto_transfer_args(self, *args, **kwargs):
71
72
def parameter_validation(fn: Callable) -> Callable:
73
"""
74
- Decorator for :meth:`~pytorch_lightning.core.LightningModule.to` method.
75
Validates that the module parameter lengths match after moving to the device. It is useful
76
when tying weights on TPU's.
77
0 commit comments