torch.utils.data — PyTorch 2.3 documentation
In the page, it says default_collate_fn_map.update({torch.Tensor: collate_tensor_fn})
can be used to update defalut_collate. For my case, I want to add default_collate_fn_map.update({PIL.Image: lambda x: return x})
. However, I cannot change default_collate_fn_map because it’s defined in torch/utils/data/_utils/collate.py. Any help?