Gagan Bhatia commited on
Commit
30f6bdb
1 Parent(s): 3315012

Update model.py

Browse files
Files changed (1) hide show
  1. 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
- """ training dataloader """
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
  )