[Bug]: Voicebank/*/*MetricGAN* torch.multinomial error

[anautsch]

Describe the bug

While testing recipes, an error was thrown for all MetricGAN recipes. This report points to one log only, but it is alike for all three MetricGAN recipes.

RuntimeError: cannot sample n_sample > prob_dist.size(-1) samples without replacement

Expected behaviour

No error

To Reproduce

No response

Versions

No response

Relevant log output

File "speechbrain/recipes/Voicebank/enhance/MetricGAN/train.py", line 371, in train_discriminator
    self.fit(
...
  File "torch/utils/data/sampler.py", line 203, in __iter__
    rand_tensor = torch.multinomial(self.weights, self.num_samples, self.replacement, generator=self.generator)
RuntimeError: cannot sample n_sample > prob_dist.size(-1) samples without replacement

Additional context

No response

[pplantinga]

This was fixed in f053563