| 787 | |
| 788 | |
| 789 | class WrappedDataLoader: |
| 790 | def __init__(self, dl, func): |
| 791 | self.dl = dl |
| 792 | self.func = func |
| 793 | |
| 794 | def __len__(self): |
| 795 | return len(self.dl) |
| 796 | |
| 797 | def __iter__(self): |
| 798 | for b in self.dl: |
| 799 | yield (self.func(*b)) |
| 800 | |
| 801 | train_dl, valid_dl = get_data(train_ds, valid_ds, bs) |
| 802 | train_dl = WrappedDataLoader(train_dl, preprocess) |