Shortcuts

DoublePrecisionPlugin

class pytorch_lightning.plugins.precision.DoublePrecisionPlugin[source]

Bases: pytorch_lightning.plugins.precision.precision_plugin.PrecisionPlugin

Plugin for training with double (torch.float64) precision.

connect(model, optimizers, lr_schedulers)[source]

Converts the model to double precision and wraps it in a LightningDoublePrecisionModule to convert incoming floating point data to double (torch.float64) precision.

Does not alter optimizers or lr_schedulers.

Return type

Tuple[Module, List[Optimizer], List[Any]]

forward_context()[source]

A context manager to change the default tensor type.

See: torch.set_default_tensor_type()

Return type

Generator[None, None, None]

Read the Docs v: latest
Versions
latest
stable
1.4.9
1.4.8
1.4.7
1.4.6
1.4.5
1.4.4
1.4.3
1.4.2
1.4.1
1.4.0
1.3.8
1.3.7
1.3.6
1.3.5
1.3.4
1.3.3
1.3.2
1.3.1
1.3.0
1.2.10
1.2.8
1.2.7
1.2.6
1.2.5
1.2.4
1.2.3
1.2.2
1.2.1
1.2.0
1.1.8
1.1.7
1.1.6
1.1.5
1.1.4
1.1.3
1.1.2
1.1.1
1.1.0
1.0.8
1.0.7
1.0.6
1.0.5
1.0.4
1.0.3
1.0.2
1.0.1
1.0.0
0.10.0
0.9.0
0.8.5
0.8.4
0.8.3
0.8.2
0.8.1
0.8.0
0.7.6
0.7.5
0.7.4
0.7.3
0.7.2
0.7.1
0.7.0
0.6.0
0.5.3
0.4.9
ipynb-update
docs-search
Downloads
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.