transforms: fix wrong gaussian blur arg

akors · May 17, 2023 · 0b70d83 · 0b70d83
1 parent 561cf03
commit 0b70d83
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/transforms.py b/transforms.py
@@ -91,7 +91,7 @@ def make_transforms(mean, std, augment_level=0):
 
         # very mild gaussian blur, almost no effect. we leave it in as a good luck charm.
         # larger kernels or larger sigma degrade performance, even on validation set.
-        oplist.append(T.GaussianBlur(kernel_size=3, sigma=(0.1,1)))
+        oplist.append(T.GaussianBlur(kernel_size=3, sigma=(0.1, 1.0)))
 
     assert augment_level <= 2, "Augmentation level "+str(augment_level)+"?? What is this, the future??"