Solved problems on newer versions of pytorch, for Windows

[SileYin]

There are some issues running this program on pytorch version>1.0 and is only on Linux/Unix but not on windows. See: https://discuss.pytorch.org/t/not-using-multiprocessing-but-getting-cuda-error-re-forked-subprocess/54610 Solved it