HI, when I used multi-gpu , I met the error as following picture shows.  how could I do to solve the problem? I think this is related with nn.DataParallel, but I'm a new newbie in pytorch.
HI,

when I used multi-gpu , I met the error as following picture shows.
how could I do to solve the problem?
I think this is related with nn.DataParallel, but I'm a new newbie in pytorch.