add inits

Co-authored-by: Adrian Wälchli <aedu.waelchli@gmail.com>
Lightning-AI · Jan 30, 2021 · c028522 · c028522
1 parent 9d18d27
commit c028522
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 0 deletions.
diff --git a/pytorch_lightning/plugins/__init__.py b/pytorch_lightning/plugins/__init__.py
@@ -0,0 +1,3 @@
+from pytorch_lightning.plugins.base_plugin import Plugin
+from pytorch_lightning.plugins.precision import *
+from pytorch_lightning.plugins.training_type import *
diff --git a/pytorch_lightning/plugins/precision/__init__.py b/pytorch_lightning/plugins/precision/__init__.py
@@ -0,0 +1,6 @@
+from pytorch_lightning.plugins.precision.apex_amp import ApexMixedPrecisionPlugin
+from pytorch_lightning.plugins.precision.mixed import MixedPrecisionPlugin
+from pytorch_lightning.plugins.precision.native_amp import NativeMixedPrecisionPlugin
+from pytorch_lightning.plugins.precision.precision_plugin import PrecisionPlugin
+from pytorch_lightning.plugins.precision.sharded_native_amp import ShardedNativeMixedPrecisionPlugin
+from pytorch_lightning.plugins.precision.tpu_bfloat import TPUHalfPrecisionPlugin