Skip to main content

create_optimizer

trident::neural::training::supervised

Function create_optimizer 

Source 
pub fn create_optimizer<B: AutodiffBackend>(
    config: &SupervisedConfig,
) -> impl Optimizer<NeuralCompilerV2<B>, B>
Expand description

Create an AdamW optimizer with gradient clipping.