Gagan Bhatia commited on
Commit
ae4f8b6
·
1 Parent(s): 68248ca

Update model.py

Browse files
Files changed (1) hide show
  1. src/models/model.py +5 -0
src/models/model.py CHANGED
@@ -136,3 +136,8 @@ class PLDataModule(LightningDataModule):
136
  """ test dataloader """
137
  return DataLoader(
138
  self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=2
 
 
 
 
 
 
136
  """ test dataloader """
137
  return DataLoader(
138
  self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=2
139
+ )
140
+
141
+ def val_dataloader(self):
142
+ """ validation dataloader """
143
+ return DataLoader(