Unverified Commit 05d76519 authored by Philip Meier's avatar Philip Meier Committed by GitHub
Browse files

enable Shuffler in Caltech prototype (#4517)


Co-authored-by: default avatarVasilis Vryniotis <datumbox@users.noreply.github.com>
parent 6d9ab576
......@@ -120,8 +120,7 @@ class Caltech101(Dataset):
images_dp = TarArchiveReader(images_dp)
images_dp = Filter(images_dp, self._is_not_background_image)
# FIXME: add this after https://github.com/pytorch/pytorch/issues/65808 is resolved
# images_dp = Shuffler(images_dp, buffer_size=INFINITE_BUFFER_SIZE)
images_dp = Shuffler(images_dp, buffer_size=INFINITE_BUFFER_SIZE)
anns_dp = TarArchiveReader(anns_dp)
anns_dp = Filter(anns_dp, self._is_ann)
......@@ -189,8 +188,7 @@ class Caltech256(Dataset):
dp = resource_dps[0]
dp = TarArchiveReader(dp)
dp = Filter(dp, self._is_not_rogue_file)
# FIXME: add this after https://github.com/pytorch/pytorch/issues/65808 is resolved
# dp = Shuffler(dp, buffer_size=INFINITE_BUFFER_SIZE)
dp = Shuffler(dp, buffer_size=INFINITE_BUFFER_SIZE)
return Mapper(dp, self._collate_and_decode_sample, fn_kwargs=dict(decoder=decoder))
def generate_categories_file(self, root: Union[str, pathlib.Path]) -> None:
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment