In the _sanitize_pytorch_types function used in petastorm dataloaders, an exception is raised when an input is a string:
|
elif re.search('[SaUO]', value.dtype.str): |
Is this really necessary? Maybe the default_collate function from PyTorch might not always like, although it seems to support it according to the documentation, but this prevents making other types of collate functions.
Would you consider removing the exception? Thank you.
In the
_sanitize_pytorch_typesfunction used in petastorm dataloaders, an exception is raised when an input is a string:petastorm/petastorm/pytorch.py
Line 64 in 0b0775a
Is this really necessary? Maybe the
default_collatefunction from PyTorch might not always like, although it seems to support it according to the documentation, but this prevents making other types of collate functions.Would you consider removing the exception? Thank you.