(worker_id)
| 14 | |
| 15 | |
| 16 | def seed_worker(worker_id): |
| 17 | worker_seed = torch.initial_seed() % 2**32 |
| 18 | np.random.seed(worker_seed) |
| 19 | random.seed(worker_seed) |
| 20 | |
| 21 | def collate(batch): |
| 22 | # Keep a numpy array on items that don't need to be a tensor for training. |
nothing calls this directly
no outgoing calls
no test coverage detected