Indentation mistake. Fixes #81

rwightman · rwightman · commit d66819d1f315 · 2020-02-04T22:56:00.000-08:00
diff --git a/timm/data/transforms_factory.py b/timm/data/transforms_factory.py
@@ -69,7 +69,7 @@ def transforms_imagenet_train(
         else:
             # if it's a scalar, duplicate for brightness, contrast, and saturation, no hue
             color_jitter = (float(color_jitter),) * 3
-            secondary_tfl += [transforms.ColorJitter(*color_jitter)]
+        secondary_tfl += [transforms.ColorJitter(*color_jitter)]
 
     final_tfl = []
     if use_prefetcher:
