Skip to content
Prev Previous commit
Next Next commit
Update loreal_poc/dataloaders/base.py
  • Loading branch information
rabah-khalek authored Jan 4, 2024
commit 9480fa94bc90a29c072ec025f67718c48ab816f4
2 changes: 0 additions & 2 deletions loreal_poc/dataloaders/base.py
Original file line number Diff line number Diff line change
Expand Up @@ -104,8 +104,6 @@ def _collate_fn(
# for key in meta_keys
# }

# if len(batched_elements[0]) != self.batch_size:
# raise StopIteration

return batched_elements

Expand Down