RuntimeError: cannot sample n_sample > prob_dist.size(-1) samples without replacement
@titu1994 @meghmak13 
Dear Team,
      I am training **conformer based SSL** model where I have configured the parameters 
**min_duration : 3.2
patch_size: 16
mask_patches: 0.5
num_negatives: 40
sample_from_same_utterance_only: true
sample_from_non_masked: false
codebook_size: 300
num_groups: 2
num_classes: 90000**

 but I am getting following error
**RuntimeError: cannot sample n_sample > prob_dist.size(-1) samples without replacement**  

I am using **Nemo-22.05 images(ngc) and 48, A100-SXM4-40GB GPU** i.e 6 nodes for the training. Kindly suggest me what is need to be done ? Thanks in advance