From 3d4bc5d8de263d313254ee1348438b2a55169eca Mon Sep 17 00:00:00 2001 From: Victor Hall Date: Sat, 17 Dec 2022 02:12:38 -0500 Subject: [PATCH] fix dl test script --- test/test_dl.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/test/test_dl.py b/test/test_dl.py index 527030d..5a28c20 100644 --- a/test/test_dl.py +++ b/test/test_dl.py @@ -6,8 +6,8 @@ import argparse import ldm.data.data_loader as dl -def main(data_root, batch_size): - data_loader = dl.DataLoaderMultiAspect(data_root=data_root, batch_size=batch_size, debug_level=1, resolution=512) +def main(data_root, batch_size, resolution): + data_loader = dl.DataLoaderMultiAspect(data_root=data_root, batch_size=batch_size, resolution=resolution, debug_level=1) image_caption_pairs = data_loader.get_all_images()