https://pytorch.org/docs/stable/notes/windows.html#multiprocessing-error-without-if-clause-protection