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
