Shared memory with torch.multiprocessing

Thanks again.

Let me also try your code on my machine. And create a minimal code implementation that I can share here.