MCPcopy Create free account
hub / github.com/tensorflow/lingvo / ScaleGradients

Method ScaleGradients

lingvo/core/learner.py:434–500  ·  view source on GitHub ↗

Scales gradients according to training params. Args: var_grads: a `.NestedMap` whose values are (var, grad) pairs. gradient_adjuster: if not None, a function that mutates a given var_grads. Returns: A `.NestedMap` containing - final_var_grads: a `.NestedMap` whose

(self, var_grads, gradient_adjuster=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Calls 4

_AddEvalMetricMethod · 0.95
_GetGlobalGradScaleMethod · 0.95
FlattenItemsMethod · 0.80
FlattenMethod · 0.45