Skip to main content

create_bert_optimizer

Function create_bert_optimizer 

Source
pub fn create_bert_optimizer(
    learning_rate: f32,
    warmup_steps: usize,
    total_steps: usize,
) -> Result<BERTOptimizer>
Expand description

Factory functions for creating task-specific optimizers