Struct burn::data::dataloader::FixBatchStrategy

pub struct FixBatchStrategy<I> { /* private fields */ }

A strategy to batch items with a fixed batch size.

Implementations

impl<I> FixBatchStrategy<I>

pub fn new(batch_size: usize) -> FixBatchStrategy<I>

Creates a new strategy to batch items with a fixed batch size.

Arguments

• batch_size - The batch size.

Returns

The strategy.

Trait Implementations

impl<I> BatchStrategy<I> for FixBatchStrategy<I>

where I: Send + 'static,

fn add(&mut self, item: I)

Adds an item to the strategy.

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

Batches the items.

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

Creates a new strategy of the same type.