From cb621e0f007c0d8e67ca5915fe5ad2e133b66114 Mon Sep 17 00:00:00 2001 From: Ross Wightman Date: Fri, 13 Aug 2021 12:54:12 -0700 Subject: [PATCH] Remove print, arg order --- timm/data/fetcher.py | 1 - timm/data/prefetcher_cuda.py | 4 ++-- 2 files changed, 2 insertions(+), 3 deletions(-) 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