BatchDataLoader

tiny_recursive_rs::data

Trait BatchDataLoader 

Source 
pub trait BatchDataLoader {
    // Required methods
    fn next_batch(
        &mut self,
        device: &Device,
    ) -> Result<Option<(Tensor, Tensor)>>;
    fn reset(&mut self);
    fn num_batches(&self) -> usize;
}
Expand description

Generic data loader trait

Required Methods§

Source

fn next_batch(&mut self, device: &Device) -> Result<Option<(Tensor, Tensor)>>

Get next batch of (input, target) tensors

Source

fn reset(&mut self)

Reset loader for new epoch

Source

fn num_batches(&self) -> usize

Get total number of batches

Implementors§