(db, queue, sample_data, data_aug)
| 35 | return args |
| 36 | |
| 37 | def prefetch_data(db, queue, sample_data, data_aug): |
| 38 | ind = 0 |
| 39 | print("start prefetching data...") |
| 40 | np.random.seed(os.getpid()) |
| 41 | while True: |
| 42 | try: |
| 43 | data, ind = sample_data(db, ind, data_aug=data_aug) |
| 44 | queue.put(data) |
| 45 | except Exception as e: |
| 46 | traceback.print_exc() |
| 47 | raise e |
| 48 | |
| 49 | def pin_memory(data_queue, pinned_data_queue, sema): |
| 50 | while True: |
nothing calls this directly
no test coverage detected