bugfix: passing args
Summary: Pull Request resolved:
@@ -228,7 +228,10 @@ class MaskedLMDataset(FairseqDataset):
# mask according to specified probabilities.
masked_blk_one, masked_tgt_one = self._mask_block(
- s["block_one"], self.mask_idx, self.pad_idx, token_range)
+ s["block_one"], self.mask_idx, self.pad_idx, token_range,
+ masking_ratio=self.masking_ratio, masking_prob=self.masking_prob,
+ random_token_prob=self.random_token_prob,
+ )
tokens = np.concatenate([
[self.classif_token_idx], masked_blk_one
