diff options
| author | junyanz <junyanzhu89@gmail.com> | 2018-04-15 00:47:21 -0400 |
|---|---|---|
| committer | junyanz <junyanzhu89@gmail.com> | 2018-04-15 00:47:21 -0400 |
| commit | 1cf7171641711d69663733989c267110d66ca466 (patch) | |
| tree | 18ed32e12bb26a2abce188ead139c8dee033d97e /data | |
| parent | 1fd836eb35924e9771253ad6fa8f74ed16a62f0b (diff) | |
fix a bug (batchSize >1) in the data_loader
Diffstat (limited to 'data')
| -rw-r--r-- | data/__init__.py | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/data/__init__.py b/data/__init__.py index ef581e7..341281d 100644 --- a/data/__init__.py +++ b/data/__init__.py @@ -49,6 +49,6 @@ class CustomDatasetDataLoader(BaseDataLoader): def __iter__(self): for i, data in enumerate(self.dataloader): - if i >= self.opt.max_dataset_size: + if i * self.opt.batchSize >= self.opt.max_dataset_size: break yield data |