Skip to main content

Module retro_forward

burn::backend::autodiff::checkpoint

Module retro_forward

Expand description

RetroForward module

Traits§

RetroForward: Definition of the forward function of a node, called during retropropagation only. This is different from the normal forward function because it reads and writes from the BackwardStates map instead of having a clear function signature.