Struct burn::backend::autodiff::checkpoint::base::Checkpointer

pub struct Checkpointer { /* private fields */ }

Struct responsible of fetching the output for a node in the autodiff graph during a backward pass

Implementations

impl Checkpointer

pub fn new( backward_states: BackwardStates, retro_forwards: RetroForwards, node_tree: NodeTree, ) -> Checkpointer

Constructs a new Checkpointer.

impl Checkpointer

pub fn retrieve_node_output<T>(&mut self, node_id: NodeID) -> T

where T: Clone + Send + 'static,

Gives the output of the given node, by recursively asking parents to compute themselves or give their pre-computed tensors.

Trait Implementations

impl Debug for Checkpointer

fn fmt(&self, f: &mut Formatter<'_>) -> Result<(), Error>

Formats the value using the given formatter.