Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
·
cd7fa2f
1
Parent(s):
b3810dd
Update model.py
Browse files- src/models/model.py +3 -0
src/models/model.py
CHANGED
|
@@ -153,6 +153,9 @@ class PLDataModule(LightningDataModule):
|
|
| 153 |
"""validation dataloader"""
|
| 154 |
return DataLoader(
|
| 155 |
self.test_dataset,
|
|
|
|
|
|
|
|
|
|
| 156 |
)
|
| 157 |
|
| 158 |
|
|
|
|
| 153 |
"""validation dataloader"""
|
| 154 |
return DataLoader(
|
| 155 |
self.test_dataset,
|
| 156 |
+
batch_size=self.batch_size,
|
| 157 |
+
shuffle=False,
|
| 158 |
+
num_workers=self.num_workers,
|
| 159 |
)
|
| 160 |
|
| 161 |
|