Open wd1900 opened 7 years ago
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.
model.module.encode_w
@UdonDa could you please expand your answer? where to fix the code so it would work on multiple gpu?\ thanx
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.