Gagan Bhatia commited on
Commit
39c6b69
1 Parent(s): 16c4212

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +1 -1
src/models/model.py CHANGED
@@ -142,7 +142,7 @@ class PLDataModule(LightningDataModule):
142
  def val_dataloader(self):
143
  """ validation dataloader """
144
  return DataLoader(
145
- self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=2
146
  )
147
 
148
 
 
142
  def val_dataloader(self):
143
  """ validation dataloader """
144
  return DataLoader(
145
+ self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=self.num_workers
146
  )
147
 
148