Spaces:
Runtime error
Runtime error
Gagan Bhatia commited on
Commit ·
b3810dd
1
Parent(s): 3b5d713
Update model.py
Browse files- src/models/model.py +2 -2
src/models/model.py
CHANGED
|
@@ -150,9 +150,9 @@ class PLDataModule(LightningDataModule):
|
|
| 150 |
)
|
| 151 |
|
| 152 |
def val_dataloader(self):
|
| 153 |
-
"""
|
| 154 |
return DataLoader(
|
| 155 |
-
self.test_dataset,
|
| 156 |
)
|
| 157 |
|
| 158 |
|
|
|
|
| 150 |
)
|
| 151 |
|
| 152 |
def val_dataloader(self):
|
| 153 |
+
"""validation dataloader"""
|
| 154 |
return DataLoader(
|
| 155 |
+
self.test_dataset,
|
| 156 |
)
|
| 157 |
|
| 158 |
|