TypeError: can't pickle Environment objects

sophos / SOREL-20M

Sophos-ReversingLabs 20 million sample dataset

Apache License 2.0

637 stars 132 forks source link

TypeError: can't pickle Environment objects #3

Open alriyami opened 3 years ago

alriyami commented 3 years ago

I am using windows 10 with GPU to run the experiment. I keep getting the error (TypeError: can't pickle Environment objects) it seems the issue only in Windows. the only word around here is to set workers to 0. The problem here will be the speed. the experiment will run very slow. Any other solution ?

rharang commented 3 years ago

There appears to be a workaround here: https://github.com/pytorch/vision/issues/689#issuecomment-787215916 unfortunately I don't have access to a windows 10 GPU system to test it; if you can get it to work I'd be happy to try and merge a fix in.