From e6858e35f0a08c6139c133122d222d0d85e8005d Mon Sep 17 00:00:00 2001 From: junyanz Date: Mon, 12 Jun 2017 23:52:56 -0700 Subject: update dataset mode --- data/data_loader.py | 9 ++------- 1 file changed, 2 insertions(+), 7 deletions(-) (limited to 'data/data_loader.py') 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 -- cgit v1.2.3-70-g09d2