Assigning weights to individual datasets in ConcatDataset

[ptrblck]

You could use a WeightedRandomSampler as described in this post.