Spaces:
Runtime error
Runtime error
Gagan Bhatia
commited on
Commit
•
30f6bdb
1
Parent(s):
3315012
Update model.py
Browse files- src/models/model.py +1 -1
src/models/model.py
CHANGED
@@ -132,7 +132,7 @@ class PLDataModule(LightningDataModule):
|
|
132 |
)
|
133 |
|
134 |
def train_dataloader(self):
|
135 |
-
"""
|
136 |
return DataLoader(
|
137 |
self.train_dataset, batch_size=self.batch_size, shuffle=True, num_workers=self.num_workers
|
138 |
)
|
|
|
132 |
)
|
133 |
|
134 |
def train_dataloader(self):
|
135 |
+
"""training dataloader"""
|
136 |
return DataLoader(
|
137 |
self.train_dataset, batch_size=self.batch_size, shuffle=True, num_workers=self.num_workers
|
138 |
)
|