CPU prefetcher. Args: loader: Dataloader.
| 61 | |
| 62 | |
| 63 | class CPUPrefetcher(): |
| 64 | """CPU prefetcher. |
| 65 | |
| 66 | Args: |
| 67 | loader: Dataloader. |
| 68 | """ |
| 69 | |
| 70 | def __init__(self, loader): |
| 71 | self.ori_loader = loader |
| 72 | self.loader = iter(loader) |
| 73 | |
| 74 | def next(self): |
| 75 | try: |
| 76 | return next(self.loader) |
| 77 | except StopIteration: |
| 78 | return None |
| 79 | |
| 80 | def reset(self): |
| 81 | self.loader = iter(self.ori_loader) |
| 82 | |
| 83 | |
| 84 | class CUDAPrefetcher(): |