Skip to main content

scale_gradients

Function scale_gradients 

Source
pub fn scale_gradients(
    graph: &mut Graph,
    nodes: &[NodeId],
    scale: f32,
) -> Result<(), ModelError>
Expand description

Scales gradients of the given nodes by a scalar factor.

Nodes without computed gradients are skipped.