Trait burn_core::data::dataloader::BatchStrategy

pub trait BatchStrategy<I>: Send + Sync {
    // Required methods
    fn add(&mut self, item: I);
    fn batch(&mut self, force: bool) -> Option<Vec<I>>;
    fn new_like(&self) -> Box<dyn BatchStrategy<I>>;
}

A strategy to batch items.

Required Methods

fn add(&mut self, item: I)

Adds an item to the strategy.

Arguments

• item - The item to add.

fn batch(&mut self, force: bool) -> Option<Vec<I>>

Batches the items.

Arguments

• force - Whether to force batching.

Returns

The batched items.

fn new_like(&self) -> Box<dyn BatchStrategy<I>>

Creates a new strategy of the same type.

Returns

The new strategy.

Implementors

impl<I: Send + Sync + 'static> BatchStrategy<I> for FixBatchStrategy<I>