Enabled drawing samples from different chunks instead of retrieving a…

…ll from the same chunk until depleted
TheJacksonLaboratory · Oct 11, 2024 · 349d2b5 · 349d2b5
1 parent 8ebf236
commit 349d2b5
Show file tree

Hide file tree

Showing 2 changed files with 2 additions and 2 deletions.
diff --git a/src/napari_activelearning/_acquisition.py b/src/napari_activelearning/_acquisition.py
@@ -246,7 +246,7 @@ def fine_tune(self, dataset_metadata_list: Iterable[
             dataset = zds.ZarrDataset(
                 list(dataset_metadata.values()),
                 return_positions=False,
-                draw_same_chunk=True,
+                draw_same_chunk=False,
                 patch_sampler=StaticPatchSampler(
                     patch_size=patch_sizes,
                     top_lefts=top_lefts,

diff --git a/src/napari_activelearning/_utils.py b/src/napari_activelearning/_utils.py
@@ -249,7 +249,7 @@ def get_dataloader(
     train_dataset = zds.ZarrDataset(
         list(dataset_metadata.values()),
         return_positions=True,
-        draw_same_chunk=True,
+        draw_same_chunk=False,
         patch_sampler=patch_sampler,
         shuffle=shuffle
     )