SUMIN/HuggingFace_transformer
1
0
Fork 0
Code Issues Pull Requests Actions 7 Packages Projects Releases Wiki Activity

clearly indicate shuffle=False (#6312)

Browse Source 
* Clarify shuffle

* clarify shuffle

Co-authored-by: Kevin Canwen Xu <canwenxu@126.com>
This commit is contained in:
xujiaze13
2020-08-30 04:26:10 -07:00
committed by GitHub
parent 0eecaceac7
commit 32fe44086c
1 changed files with 2 additions and 2 deletions
Download Patch File Download Diff File Expand all files Collapse all files

4
examples/lightning_base.py
View File

@@ -178,10 +178,10 @@ class BaseTransformer(pl.LightningModule):
return self.train_loader return self.train_loader
def val_dataloader(self): def val_dataloader(self):
return self.get_dataloader("dev", self.hparams.eval_batch_size) return self.get_dataloader("dev", self.hparams.eval_batch_size, shuffle=False)
def test_dataloader(self): def test_dataloader(self):
return self.get_dataloader("test", self.hparams.eval_batch_size) return self.get_dataloader("test", self.hparams.eval_batch_size, shuffle=False)
def _feature_file(self, mode): def _feature_file(self, mode):
return os.path.join( return os.path.join(