Python DataLoader.__code__ Examples

Example #1

 def to_dataloader(self,
                   batch_size,
                   num_workers=0,
                   shuffle=False,
                   pin_memory=False):
     dataloader = DataLoader(self, batch_size=batch_size, num_workers=num_workers,\
                        drop_last=False, collate_fn = self.collate, shuffle=shuffle, pin_memory=pin_memory)
     dataloader.__code__ = 0
     return dataloader

Example #2

 def to_dataloader(self, batch_size, num_workers=4, shuffle=True):
     dataloader = DataLoader(self, batch_size=batch_size, num_workers=num_workers,\
                        drop_last=False, collate_fn = self.collate, shuffle=shuffle)
     dataloader.__code__ = 0
     return dataloader