Skip to content

Commit 94e0b28

Browse files
committed
Update precision_plugin.py
1 parent 791fe19 commit 94e0b28

File tree

1 file changed

+3
-0
lines changed

1 file changed

+3
-0
lines changed

pytorch_lightning/plugins/precision/precision_plugin.py

Lines changed: 3 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -22,6 +22,9 @@
2222

2323

2424
class PrecisionPlugin(Plugin):
25+
""" Plugin handling the precision-specific parts of the training.
26+
The static classattributes EPSILON and precision must be overwritten in child-classes and their default values reflect fp32 training
27+
"""
2528
EPSILON = 1e-6
2629
precision = 32
2730

0 commit comments

Comments
 (0)