Remove useless comma

This commit is contained in:
henryruhs
2025-03-04 12:17:59 +01:00
parent df5895e266
commit fea75ff949
+1 -2
View File
@@ -86,7 +86,7 @@ class AugmentTransform:
albumentations.MotionBlur(p = 0.1),
albumentations.MedianBlur(p = 0.1)
], p = 0.3),
albumentations.ColorJitter(p = 0.1),
albumentations.ColorJitter(p = 0.1)
])
@@ -97,4 +97,3 @@ class WarpTransform:
def __call__(self, input_tensor : Tensor) -> Tensor:
temp_tensor = input_tensor.unsqueeze(0)
return warp_tensor(temp_tensor, self.warp_template).squeeze(0)