Remove print, arg order

pull/1239/head
Ross Wightman 3 years ago
parent b974d85026
commit cb621e0f00

@ -51,7 +51,6 @@ class Fetcher:
self._loader = pl.MpDeviceLoader(loader, device) self._loader = pl.MpDeviceLoader(loader, device)
else: else:
self._loader = loader self._loader = loader
print('re', self.random_erasing, self.mean, self.std)
def __iter__(self): def __iter__(self):
for sample, target in self._loader: for sample, target in self._loader:

@ -16,10 +16,10 @@ class PrefetcherCuda:
normalize_shape=(1, 3, 1, 1), normalize_shape=(1, 3, 1, 1),
mean=IMAGENET_DEFAULT_MEAN, mean=IMAGENET_DEFAULT_MEAN,
std=IMAGENET_DEFAULT_STD, std=IMAGENET_DEFAULT_STD,
num_aug_splits=0,
re_prob=0., re_prob=0.,
re_mode='const', re_mode='const',
re_count=1 re_count=1,
num_aug_splits=0,
): ):
self.loader = loader self.loader = loader
self.device = device self.device = device

Loading…
Cancel
Save