Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
3b5d713
1
Parent(s):
2c57b14
Update model.py
Browse files- src/models/model.py +4 -1
src/models/model.py
CHANGED
@@ -143,7 +143,10 @@ class PLDataModule(LightningDataModule):
|
|
143 |
def test_dataloader(self):
|
144 |
"""test dataloader"""
|
145 |
return DataLoader(
|
146 |
-
self.test_dataset,
|
|
|
|
|
|
|
147 |
)
|
148 |
|
149 |
def val_dataloader(self):
|
|
|
143 |
def test_dataloader(self):
|
144 |
"""test dataloader"""
|
145 |
return DataLoader(
|
146 |
+
self.test_dataset,
|
147 |
+
batch_size=self.batch_size,
|
148 |
+
shuffle=False,
|
149 |
+
num_workers=self.num_workers,
|
150 |
)
|
151 |
|
152 |
def val_dataloader(self):
|