-record(data_loader, {
    dataset :: viva_tensor@data@dataloader:dataset(),
    batch_size :: integer(),
    shuffle :: boolean(),
    drop_last :: boolean()
}).