docs

Lightning-AI · Oct 26, 2021 · 6d3f60e · 6d3f60e
1 parent 92bd400
commit 6d3f60e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/docs/source/extensions/accelerators.rst b/docs/source/extensions/accelerators.rst
@@ -26,7 +26,7 @@ One to handle differences from the training routine and one to handle different
     from pytorch_lightning.plugins import NativeMixedPrecisionPlugin, DDPPlugin
 
     accelerator = GPUAccelerator(
-        precision_plugin=NativeMixedPrecisionPlugin(...),
+        precision_plugin=NativeMixedPrecisionPlugin(16, "cuda"),
         training_type_plugin=DDPPlugin(),
     )
     trainer = Trainer(accelerator=accelerator)