fix initial crop size

mehta-lab · Jul 31, 2024 · ddeacb8 · ddeacb8
1 parent a2c25b6
commit ddeacb8
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/viscy/data/triplet.py b/viscy/data/triplet.py
@@ -250,7 +250,7 @@ def _setup_fit(self, dataset_settings: dict):
         self.train_dataset = TripletDataset(
             positions=positions,
             tracks_tables=tracks_tables,
-            initial_yx_patch_size=self.yx_patch_size,
+            initial_yx_patch_size=self.initial_yx_patch_size,
             anchor_transform=no_aug_transform,
             positive_transform=augment_transform,
             negative_transform=augment_transform,
@@ -264,7 +264,7 @@ def _setup_predict(self, dataset_settings: dict):
         self.predict_dataset = TripletDataset(
             positions=positions,
             tracks_tables=tracks_tables,
-            initial_yx_patch_size=self.yx_patch_size,
+            initial_yx_patch_size=self.initial_yx_patch_size,
             anchor_transform=Compose(self.normalizations),
             fit=False,
             **dataset_settings,