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 the training_step, validation_step, test_step, predict_step, and forward methods to convert incoming floating point data to double. Does not alter optimizers or lr_schedulers.

Return type

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

post_dispatch()[source]

Hook to do something after the training/evaluation/prediction finishes.

Return type

None

predict_step_context()[source]

A context manager to change the default tensor type. See: torch.set_default_tensor_type()

Return type

Generator[None, None, None]

test_step_context()[source]

A context manager to change the default tensor type. See: torch.set_default_tensor_type()

Return type

Generator[None, None, None]

train_step_context()[source]

A context manager to change the default tensor type. See: torch.set_default_tensor_type()

Return type

Generator[None, None, None]

val_step_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.3.1
1.3.0
1.2.10
1.2.9_a
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
docs-robots
Downloads
pdf
html
On Read the Docs
Project Home
Builds

Free document hosting provided by Read the Docs.