Accidentally removed function

[cosmina98]

Hey guys, I got an error during training. Looks like you accidentally removed: def distribute_sampler(sampler, world_size, rank): vec = np.array(sorted(sampler)) idx = [i for i in range(len(sampler)) if i % world_size == rank] sampler = vec[idx].tolist() random.shuffle(sampler) return sampler on misc_utils.py after the last commit on that file:)