1 files changed, 1 insertions, 1 deletions
diff --git a/test.py b/test.py
index 1ddb894..cccb0c6 100644
--- a/test.py
+++ b/test.py
@@ -13,7 +13,7 @@ if __name__ == '__main__':
     opt.serial_batches = True  # no shuffle
     opt.no_flip = True  # no flip
 
-    data_loader = CreateDataLoader(opt)
+    data_loader = CreateRecursiveDataLoader(opt)
     dataset = data_loader.load_data()
     model = create_model(opt)
     visualizer = Visualizer(opt)