diff --git a/examples/fairseq/tasks/pretraining.py b/examples/fairseq/tasks/pretraining.py index 57b0628..2d32127 100644 --- a/examples/fairseq/tasks/pretraining.py +++ b/examples/fairseq/tasks/pretraining.py @@ -180,6 +180,7 @@ class PLMTask(FairseqTask): epoch=1, data_buffer_size=0, disable_iterator_cache=False, + **kwargs, ): return MLMLoader( self.cfg,