Struct GenericDataLoader 

pub struct GenericDataLoader<D, C, T>
where
    D: Dataset<Item = T>,
    C: Collate<T>,
    T: Send,
{ /* private fields */ }

A more flexible DataLoader that works with any Dataset and Collate function.

Implementations

impl<D, C, T> GenericDataLoader<D, C, T>

where D: Dataset<Item = T>, C: Collate<T>, T: Send,

pub fn new(dataset: D, collate_fn: C, batch_size: usize) -> Self

Creates a new GenericDataLoader.

pub fn shuffle(self, shuffle: bool) -> Self

Enables or disables shuffling.

pub fn drop_last(self, drop_last: bool) -> Self

Sets whether to drop the last incomplete batch.

pub fn len(&self) -> usize

Returns the number of batches.

pub fn is_empty(&self) -> bool

Returns true if empty.

pub fn iter(&self) -> GenericDataLoaderIter<'_, D, C, T>

Creates an iterator over batches.