Convert to floats at the beginning.

Author: datumbox

references/classification/presets.py

@@ -19,6 +19,7 @@ def __init__(
     ):
         trans = [
             transforms.ToImageTensor(),
+            transforms.ConvertImageDtype(torch.float),
             transforms.RandomResizedCrop(crop_size, interpolation=interpolation, antialias=True),
         ]
         if hflip_prob > 0:
@@ -35,7 +36,6 @@ def __init__(
                 trans.append(transforms.AutoAugment(policy=aa_policy, interpolation=interpolation))
         trans.extend(
             [
-                transforms.ConvertImageDtype(torch.float),
                 transforms.Normalize(mean=mean, std=std),
             ]
         )
@@ -62,9 +62,9 @@ def __init__(
         self.transforms = transforms.Compose(
             [
                 transforms.ToImageTensor(),
+                transforms.ConvertImageDtype(torch.float),
                 transforms.Resize(resize_size, interpolation=interpolation, antialias=True),
                 transforms.CenterCrop(crop_size),
-                transforms.ConvertImageDtype(torch.float),
                 transforms.Normalize(mean=mean, std=std),
             ]
         )