quant/data/data_loaders.py [173:181]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            batch_size=self.train_batch_size,
            shuffle=True,
            num_workers=self.workers,
            pin_memory=True,
        )

        return train_loader

    def get_test_loader(self) -> DataLoader:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



quant/data/data_loaders.py [343:351]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
            batch_size=self.train_batch_size,
            shuffle=True,
            num_workers=self.workers,
            pin_memory=True,
        )

        return train_loader

    def get_test_loader(self) -> DataLoader:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -