From 7f331d883622726e66b9255c3e95abe6115f951c Mon Sep 17 00:00:00 2001 From: Francis Lata Date: Tue, 21 Jan 2025 13:43:59 -0800 Subject: [PATCH] fix dataloader script --- examples/mlperf/dataloader.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/examples/mlperf/dataloader.py b/examples/mlperf/dataloader.py index cad8cf7f09..a9bda998e6 100644 --- a/examples/mlperf/dataloader.py +++ b/examples/mlperf/dataloader.py @@ -507,7 +507,7 @@ if __name__ == "__main__": dataset = COCO(download_dataset(base_dir:=getenv("BASE_DIR", BASEDIR), "validation" if val else "train")) anchors = np.ones((120087, 4)) with tqdm(total=len(dataset.imgs.keys())) as pbar: - for x in batch_load_retinanet(dataset, val, anchors, base_dir): + for x in batch_load_retinanet(dataset, val, base_dir): pbar.update(x[0].shape[0]) load_fn_name = f"load_{getenv('MODEL', 'resnet')}"