From b202ddfda2a01da2fa4ed87ca89f82cb7892bc5b Mon Sep 17 00:00:00 2001
From: Justus Schock <12886177+justusschock@users.noreply.github.com>
Date: Sun, 31 Jan 2021 11:34:26 +0100
Subject: [PATCH] Update precision_plugin.py

---
 pytorch_lightning/plugins/precision/precision_plugin.py | 3 +++
 1 file changed, 3 insertions(+)

diff --git a/pytorch_lightning/plugins/precision/precision_plugin.py b/pytorch_lightning/plugins/precision/precision_plugin.py
index 22c1a7539fff4e..031b5887376147 100644
--- 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