mrq/torchscale
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Support latest fairseq

Browse Source
This commit is contained in:
shumingma 2022-12-15 03:44:15 -08:00
parent 2518ea030c
commit 7e12b582e4
1 changed files with 1 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
examples/fairseq/tasks/pretraining.py
View File

@ -180,6 +180,7 @@ class PLMTask(FairseqTask):
epoch=1, epoch=1,
data_buffer_size=0, data_buffer_size=0,
disable_iterator_cache=False, disable_iterator_cache=False,
**kwargs,
): ):
return MLMLoader( return MLMLoader(
self.cfg, self.cfg,