def build_loader(self, batch_size, num_workers, shuffle=False): loader = DataLoader(self, batch_size=batch_size, shuffle=shuffle, num_workers=num_workers, pin_memory=True) loader.cat_feat_origin_cards = self.cat_feat_origin_cards return loader
def build_loader(self, shuffle=False): try_import_torch() from torch.utils.data import DataLoader loader = DataLoader(self, batch_size=self.kwargs['batch_size'], shuffle=shuffle, num_workers=16, pin_memory=True) loader.cat_feat_origin_cards = self.cat_feat_origin_cards return loader