custome collate function cannot get _use_shared_memory

[cy69855522]

Thx a lot for such a wonderful package~
But when I wrote a custom function for the Dataloader's collate_fn, I found I cannot get the value of variable - “_use_shared_memory” in torch.utils.data._utils.collate.py
ModuleNotFoundError: No module named 'torch.utils.data._utils'
I was thinking if you could give me some suggestions. I don't want to change the original code of pytorch locally.

[soumith]

cc: @ssnl

[ssnl]

Let’s expose that! I’ll add this field to the worker info in the upcoming PR.

[cy69855522]

Let’s expose that! I’ll add this field to the worker info in the upcoming PR.

Thx a lot~

[ssnl]

you can use torch.utils.data.get_worker_info() != None to get the same information (whether you are in a worker process) after #19228