Gagan Bhatia commited on
Commit
cd7fa2f
1 Parent(s): b3810dd

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +3 -0
src/models/model.py CHANGED
@@ -153,6 +153,9 @@ class PLDataModule(LightningDataModule):
153
  """validation dataloader"""
154
  return DataLoader(
155
  self.test_dataset,
 
 
 
156
  )
157
 
158
 
 
153
  """validation dataloader"""
154
  return DataLoader(
155
  self.test_dataset,
156
+ batch_size=self.batch_size,
157
+ shuffle=False,
158
+ num_workers=self.num_workers,
159
  )
160
 
161