Skip to main content

clip_gradient

Function clip_gradient 

Source
pub fn clip_gradient(grad: &[f64], max_norm: f64) -> Vec<f64>
Expand description

Clip gradient to max norm (per-sample)