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.
2 parents 7c51ab7 + 31fe222 commit 9dc0962Copy full SHA for 9dc0962
docs/source/extensions/accelerators.rst
@@ -26,8 +26,8 @@ One to handle differences from the training routine and one to handle different
26
from pytorch_lightning.plugins import NativeMixedPrecisionPlugin, DDPPlugin
27
28
accelerator = GPUAccelerator()
29
- precision_plugin=NativeMixedPrecisionPlugin(precision=16, device="cuda")
30
- training_type_plugin=DDPPlugin(accelerator=accelerator, precision_plugin=precision_plugin)
+ precision_plugin = NativeMixedPrecisionPlugin(precision=16, device="cuda")
+ training_type_plugin = DDPPlugin(accelerator=accelerator, precision_plugin=precision_plugin)
31
trainer = Trainer(strategy=training_type_plugin)
32
33
0 commit comments