Fix typo in docstring (#5835)
Co-authored-by: mergify[bot] <37929162+mergify[bot]@users.noreply.github.com>
(cherry picked from commit 0a50bb406f
)
This commit is contained in:
parent
cb67e1d0b2
commit
f20c1d1cdb
|
@ -107,7 +107,7 @@ class ApexPlugin(PrecisionPlugin):
|
|||
grad_clip_val: Maximum norm of gradients.
|
||||
optimizer: Optimizer with gradients that will be clipped.
|
||||
norm_type: (float or int): type of the used p-norm. Can be ``'inf'`` for
|
||||
infinity norm.
|
||||
infinity norm.
|
||||
"""
|
||||
model = self.trainer.get_model()
|
||||
parameters = model.parameters()
|
||||
|
|
Loading…
Reference in New Issue