Skip to main content

HasAutodiffModule

burn_core::module

Trait HasAutodiffModule 

Source 
pub trait HasAutodiffModule<B: AutodiffBackend> {
    type TrainModule: AutodiffModule<B, InnerModule = Self>;
}
Expand description

Helper trait to associate a module with its autodiff version.

Required Associated Types§

Source

type TrainModule: AutodiffModule<B, InnerModule = Self>

The module with auto-differentiation.

Implementors§