Expand description
Gradient centralization for training stabilization
Functionsยง
- centralize_
gradients - Apply gradient centralization to gradients
- centralize_
gradients_ with_ scaling - Apply gradient centralization with scaling
- centralize_
single_ gradient - Apply gradient centralization to a single gradient array