NVIDIA/DeepRecommender

'DataParallel' object has no attribute 'encode_w'

wd1900 opened this issue · 2 comments

wd1900 commented

HI,
when I used multi-gpu , I met the error as following picture shows.
image

how could I do to solve the problem?

I think this is related with nn.DataParallel, but I'm a new newbie in pytorch.

UdonDa commented 
model.module.encode_w
  • 👍4
  • ❤️1
mlcoop commented

@UdonDa could you please expand your answer?
where to fix the code so it would work on multiple gpu?
thanx

  • 👍1