Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
39c6b69
1
Parent(s):
16c4212
Update model.py
Browse files- src/models/model.py +1 -1
src/models/model.py
CHANGED
@@ -142,7 +142,7 @@ class PLDataModule(LightningDataModule):
|
|
142 |
def val_dataloader(self):
|
143 |
""" validation dataloader """
|
144 |
return DataLoader(
|
145 |
-
self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=
|
146 |
)
|
147 |
|
148 |
|
|
|
142 |
def val_dataloader(self):
|
143 |
""" validation dataloader """
|
144 |
return DataLoader(
|
145 |
+
self.test_dataset, batch_size=self.batch_size, shuffle=False, num_workers=self.num_workers
|
146 |
)
|
147 |
|
148 |
|