Update precision_plugin.py

Lightning-AI · Jan 31, 2021 · b202ddf · b202ddf
1 parent c028522
commit b202ddf
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