Fix num_magnitude_bins default in TrivialAugment

SamuelGabriel · Sep 6, 2021 · 9df8f13 · 9df8f13
1 parent 30bbae9
commit 9df8f13
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/torchvision/transforms/autoaugment.py b/torchvision/transforms/autoaugment.py
@@ -345,7 +345,7 @@ class TrivialAugmentWide(torch.nn.Module):
             image. If given a number, the value is used for all bands respectively.
         """
 
-    def __init__(self, num_magnitude_bins: int = 30, interpolation: InterpolationMode = InterpolationMode.NEAREST,
+    def __init__(self, num_magnitude_bins: int = 31, interpolation: InterpolationMode = InterpolationMode.NEAREST,
                  fill: Optional[List[float]] = None) -> None:
         super().__init__()
         self.num_magnitude_bins = num_magnitude_bins