def my_worker_init_fn(worker_id): np.random.seed(np.random.get_state()[1][0] + worker_id) pass

这是一个用于 PyTorch DataLoader 的 worker 初始化函数，可以用于设置每个 worker 的随机种子，以确保每个 worker 使用不同的数据。该函数使用 NumPy 的随机种子，将其设置为当前随机状态的第一个值加上 worker_id。这样，每个 worker 就可以使用一个不同的随机种子来获取不同的数据。