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 |
|