eazygrad.data.dataloader.Dataloader

class eazygrad.data.dataloader.Dataloader(dataset, batch_size, shuffle=True, drop_last=True)[source]

Bases: object

Very simple dataloader with no multiprocessing (mostly for MNIST which is already loaded in RAM)