I would like to update WeightedRandomSampler.weights between epochs, e.g. for curriculum learning. A look at its class implementation suggests that I can just update the class field “weights” in b/w epochs, because the __iter__ method simply draws from the multinomial distribution - with self.weight as input. Is that correct, or are their broader implications that I’m missing, e.g. something in the Sampler class?