(self, loader, opt)
| 95 | """ |
| 96 | |
| 97 | def __init__(self, loader, opt): |
| 98 | self.ori_loader = loader |
| 99 | self.loader = iter(loader) |
| 100 | self.opt = opt |
| 101 | self.stream = torch.cuda.Stream() |
| 102 | self.device = torch.device('cuda' if opt['num_gpu'] != 0 else 'cpu') |
| 103 | self.preload() |
| 104 | |
| 105 | def preload(self): |
| 106 | try: |