Indentation mistake. Fixes #81

This commit is contained in:
Ross Wightman 2020-02-04 22:56:00 -08:00
parent 65cda1c63f
commit d66819d1f3

View File

@ -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: