Trait burn_core::data::dataloader::DataLoader

pub trait DataLoader<O> {
    // Required methods
    fn iter<'a>(&'a self) -> Box<dyn DataLoaderIterator<O> + 'a>;
    fn num_items(&self) -> usize;
}

A data loader that can be used to iterate over a dataset.

Required Methods

fn iter<'a>(&'a self) -> Box<dyn DataLoaderIterator<O> + 'a>

Returns a boxed iterator to iterate over the data loader.

fn num_items(&self) -> usize

The number of items (not the number of batches nor the number of iterations), corresponding to the items_total of the progress returned by the iterator.

Implementors

impl<I: Send + Sync + Clone + 'static, O: Send + Sync> DataLoader<O> for BatchDataLoader<I, O>

impl<O> DataLoader<O> for MultiThreadDataLoader<O>

where O: Send + 'static + Debug,