Thanks for your reply!
But my use case (I’m trying model-based data augmentation) strictly constraints that each sample is processed differently.
In this case, as I can’t process the samples in batch-wise, I believe that running multiprocessing is the last hope for me ;(