nntoolbox.callbacks.gradient module

class nntoolbox.callbacks.gradient.GradientNormClipping(max_norm: float, norm_type=2)[source]

Bases: nntoolbox.callbacks.callbacks.Callback

after_backward() → bool[source]
class nntoolbox.callbacks.gradient.GradientValueClipping(clip_value: float)[source]

Bases: nntoolbox.callbacks.callbacks.Callback

after_backward() → bool[source]