diff --git a/pytorch_lightning/plugins/training_type/deepspeed.py b/pytorch_lightning/plugins/training_type/deepspeed.py index f04da82e04980..75e5bf74be643 100644 --- 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)