-
Notifications
You must be signed in to change notification settings - Fork 5
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
pytorch on windows does not support multiprocessing #6
Comments
Wrapping the main calls at the bottom of pggan.py in main seems to have solved the multiprocessing problem
|
Also appears you have to remove Volatile=true since that's been deprecated |
In addition all data[0] references need to be switched to item() which is a result of the Volatile tag being deprecated |
The last thing I had to do to get this working on Windows 10 pytorch was change all the directory and file creation paths to use backslashes instead of forward slashes. I also have to remove the '-p' flag in mkdir calls. I commented out the tensorboard code section until I can spend some time installing it. So far it seems to be running well on my Dell XPS with an nVidia 1080 (6GB). |
Also had to add in the following modification because DataParallel puts the original object into the module variable
|
In intermediate_block you have to cast the ndim as an int because in python3 dividing two integers results in a float. grrr only collided into this one when the resl got to 6 and the code to divide ndim by two clicked in.
|
I'm attempting to figure out which bits of the code attempt to spawn new processes since this fails on Windows 10 pytorch. I had to modify pcgan in a similar manner to get it to work recently.
The text was updated successfully, but these errors were encountered: