pub fn train_step_adam(
graph: &mut Graph,
optimizer: &mut Adam,
prediction: NodeId,
target: NodeId,
trainable_nodes: &[NodeId],
) -> Result<f32, ModelError>Expand description
Runs one full train step: loss forward, backward, and Adam updates.