pub struct DataLoader<D>where
D: Dataset,{ /* private fields */ }Implementations§
Source§impl<D> DataLoader<D>where
D: Dataset,
impl<D> DataLoader<D>where
D: Dataset,
pub fn new(dataset: D, batch_size: usize) -> DataLoader<D>
pub fn new_with_config( dataset: D, batch_size: usize, shuffle: bool, drop_last: bool, collate_fn: fn(Vec<DataItem>) -> Result<Batch, HoduError>, seed: Option<u64>, ) -> DataLoader<D>
pub fn builder(dataset: D) -> DataLoaderBuilder<D>
pub fn batch_size(&self) -> usize
pub fn num_batches(&self) -> usize
pub fn reset(&mut self)
pub fn has_next(&self) -> bool
pub fn next_batch(&mut self) -> Result<Option<Batch>, HoduError>
pub fn iter_batches(&mut self) -> DataLoaderIterator<'_, D> ⓘ
Auto Trait Implementations§
impl<D> Freeze for DataLoader<D>where
D: Freeze,
impl<D> RefUnwindSafe for DataLoader<D>where
D: RefUnwindSafe,
impl<D> Send for DataLoader<D>where
D: Send,
impl<D> Sync for DataLoader<D>where
D: Sync,
impl<D> Unpin for DataLoader<D>where
D: Unpin,
impl<D> UnsafeUnpin for DataLoader<D>where
D: UnsafeUnpin,
impl<D> UnwindSafe for DataLoader<D>where
D: UnwindSafe,
Blanket Implementations§
Source§impl<T> BorrowMut<T> for Twhere
T: ?Sized,
impl<T> BorrowMut<T> for Twhere
T: ?Sized,
Source§fn borrow_mut(&mut self) -> &mut T
fn borrow_mut(&mut self) -> &mut T
Mutably borrows from an owned value. Read more