Skip to main content

BatchUpdate

trustformers_optim::parallel

Trait BatchUpdate 

Source 
pub trait BatchUpdate {
    // Required method
    fn update_batch(&mut self, batch: Vec<(&mut Tensor, &Tensor)>) -> Result<()>;
}
Expand description

Batch parameter update interface for better parallelization.

Required Methods§

Source

fn update_batch(&mut self, batch: Vec<(&mut Tensor, &Tensor)>) -> Result<()>

Updates multiple parameters in a single batch operation.

Implementors§