fix config syntax in code

SainsburyWellcomeCentre · Mar 27, 2024 · f2c481f · f2c481f
1 parent dbdb98a
commit f2c481f
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/crabs/detection_tracking/datamodules.py b/crabs/detection_tracking/datamodules.py
@@ -55,8 +55,8 @@ def _get_train_transform(self) -> torchvision.transforms:
 
             if self.config["gaussian_blur"]:
                 gauss = transforms.GaussianBlur(
-                    kernel_size=self.config["kernel_size"],
-                    sigma=self.config["sigma"],
+                    kernel_size=self.config["gaussian_blur"]["kernel_size"],
+                    sigma=self.config["gaussian_blur"]["sigma"],
                 )
                 train_data_augm.append(gauss)