Update pytorch_lightning/plugins/training_type/deepspeed.py

Co-authored-by: Carlos Mocholí <carlossmocholi@gmail.com>
Lightning-AI · Feb 21, 2021 · 5a536cf · 5a536cf
1 parent 973b43c
commit 5a536cf
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pytorch_lightning/plugins/training_type/deepspeed.py b/pytorch_lightning/plugins/training_type/deepspeed.py
@@ -136,7 +136,7 @@ def __init__(
                 0.0 results in dynamic loss scaling, otherwise static (Default: 0)
 
             initial_scale_power: Power of the initial dynamic loss scale value. Loss scale is computed
-                by 2^initial_scale_power (Default 32)
+                by ``2^initial_scale_power`` (Default: 32)
 
             loss_scale_window: Window in which to raise/lower the dynamic FP16 loss scaling value (Default: 1000)