Update precision_plugin.py

Lightning-AI · Jan 31, 2021 · 94e0b28 · 94e0b28
1 parent 791fe19
commit 94e0b28
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/pytorch_lightning/plugins/precision/precision_plugin.py b/pytorch_lightning/plugins/precision/precision_plugin.py
@@ -22,6 +22,9 @@
 
 
 class PrecisionPlugin(Plugin):
+    """ Plugin handling the precision-specific parts of the training.
+    The static classattributes EPSILON and precision must be overwritten in child-classes and their default values reflect fp32 training
+    """
     EPSILON = 1e-6
     precision = 32