Random paramters value for torchaudio augmentation

audio
1

Here is code

How can i randomly define the value of torchaudio augmentations parameters. For example time_mask_param value range is selected between 10-30

from torchvision import transforms as T
from torchaudio import transforms as TA
aug=T.Compose([
            TA.TimeMasking(time_mask_param=10),
            TA.FrequencyMasking(freq_mask_param=10),
            TA.TimeStretch(0.9),
            
            T.Resize((224,224))
            ])
2

I guess transforms.RandomChoice (from torchvision.transforms) would work and you could use it via:

transforms.RandomChoice([TA.TimeMasking(time_mask_param=10), TA.TimeMasking(time_mask_param=20)], p=[0.5, 0.5])