diff options
Diffstat (limited to 'data/data_loader.py')
| -rw-r--r-- | data/data_loader.py | 9 |
1 files changed, 2 insertions, 7 deletions
diff --git a/data/data_loader.py b/data/data_loader.py index 69035ea..2a4433a 100644 --- a/data/data_loader.py +++ b/data/data_loader.py @@ -1,12 +1,7 @@ def CreateDataLoader(opt): - data_loader = None - if opt.align_data > 0: - from data.aligned_data_loader import AlignedDataLoader - data_loader = AlignedDataLoader() - else: - from data.unaligned_data_loader import UnalignedDataLoader - data_loader = UnalignedDataLoader() + from data.custom_dataset_data_loader import CustomDatasetDataLoader + data_loader = CustomDatasetDataLoader() print(data_loader.name()) data_loader.initialize(opt) return data_loader |