fix bug

tanganke · Nov 11, 2024 · 42dff55 · 42dff55
1 parent 7a1b2f9
commit 42dff55
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/fusion_bench/mixins/clip_classification.py b/fusion_bench/mixins/clip_classification.py
@@ -90,7 +90,7 @@ def get_shuffled_test_loader_iter(
         # create the dataloader
         loader = DataLoader(
             clip_dataset,
-            **self._dataloader_kwargs,
+            **dataloader_kwargs,
         )
         loader = self.fabric.setup_dataloaders(loader)
         return iter(InfiniteDataLoader(loader))