Indentation mistake. Fixes #81

5 years ago · d66819d1f3
parent 65cda1c63f
commit d66819d1f3
1 changed files with 1 additions and 1 deletions
--- 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: