diff --git a/timm/data/fetcher.py b/timm/data/fetcher.py index c833b596..42cf7912 100644 --- a/timm/data/fetcher.py +++ b/timm/data/fetcher.py @@ -51,7 +51,6 @@ class Fetcher: self._loader = pl.MpDeviceLoader(loader, device) else: self._loader = loader - print('re', self.random_erasing, self.mean, self.std) def __iter__(self): for sample, target in self._loader: diff --git a/timm/data/prefetcher_cuda.py b/timm/data/prefetcher_cuda.py index 9432df59..0b36c027 100644 --- a/timm/data/prefetcher_cuda.py +++ b/timm/data/prefetcher_cuda.py @@ -16,10 +16,10 @@ class PrefetcherCuda: normalize_shape=(1, 3, 1, 1), mean=IMAGENET_DEFAULT_MEAN, std=IMAGENET_DEFAULT_STD, - num_aug_splits=0, re_prob=0., re_mode='const', - re_count=1 + re_count=1, + num_aug_splits=0, ): self.loader = loader self.device = device