fixed cmd arg for shuffle dataset masked lm task
Summary: Pull Request resolved:
@@ -41,7 +41,7 @@ class MaskedLMTask(FairseqTask):
help='max number of total tokens over all segments'
' per sample for BERT dataset')
parser.add_argument('--break-mode', default="doc", type=str, help='mode for breaking sentence')
- parser.add_argument('--shuffle-dataset', default=False)
+ parser.add_argument('--shuffle-dataset', action='store_true', default=False)
def __init__(self, args, dictionary):
super().__init__(args)
