Gagan Bhatia commited on
Commit
b3810dd
·
1 Parent(s): 3b5d713

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +2 -2
src/models/model.py CHANGED
@@ -150,9 +150,9 @@ class PLDataModule(LightningDataModule):
150
  )
151
 
152
  def val_dataloader(self):
153
- """ validation dataloader """
154
  return DataLoader(
155
- self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=self.num_workers
156
  )
157
 
158
 
 
150
  )
151
 
152
  def val_dataloader(self):
153
+ """validation dataloader"""
154
  return DataLoader(
155
+ self.test_dataset,
156
  )
157
 
158