Revert inria batch_size

microsoft · Jul 2, 2024 · a738549 · a738549
1 parent a0376b8
commit a738549
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchgeo/datamodules/inria.py b/torchgeo/datamodules/inria.py
@@ -39,7 +39,7 @@ def __init__(
             **kwargs: Additional keyword arguments passed to
                 :class:`~torchgeo.datasets.InriaAerialImageLabeling`.
         """
-        super().__init__(InriaAerialImageLabeling, batch_size, num_workers, **kwargs)
+        super().__init__(InriaAerialImageLabeling, 1, num_workers, **kwargs)
 
         self.patch_size = _to_tuple(patch_size)