Closed An-Pan closed 6 years ago
As far as I know, python layer is computed through cpu. So forget about multi-gpu
@ewen1024 Tanks~
@An-Pan Meanwhile, I found the spatial drop taking so much time. I simply replaced it with the original verision of dropout and got nearly 2X speed and no noticeable performance drop.
Hi, I have try to training using multi-gpu . I got this error
So, does any one have experience on change the code spatial_dropout.py to use multi-gpu ?