Skip to content

Commit 9dc0962

Browse files
committed
Merge branch 'move-accelerator' of github.com:four4fish/pytorch-lightning into move-accelerator
2 parents 7c51ab7 + 31fe222 commit 9dc0962

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

docs/source/extensions/accelerators.rst

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -26,8 +26,8 @@ One to handle differences from the training routine and one to handle different
2626
from pytorch_lightning.plugins import NativeMixedPrecisionPlugin, DDPPlugin
2727

2828
accelerator = GPUAccelerator()
29-
precision_plugin=NativeMixedPrecisionPlugin(precision=16, device="cuda")
30-
training_type_plugin=DDPPlugin(accelerator=accelerator, precision_plugin=precision_plugin)
29+
precision_plugin = NativeMixedPrecisionPlugin(precision=16, device="cuda")
30+
training_type_plugin = DDPPlugin(accelerator=accelerator, precision_plugin=precision_plugin)
3131
trainer = Trainer(strategy=training_type_plugin)
3232

3333

0 commit comments

Comments
 (0)