Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
ae4f8b6
1
Parent(s):
68248ca
Update model.py
Browse files- 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(
|