RuntimeError: arguments are located on different GPUs

sergeytulyakov / mocogan

MoCoGAN: Decomposing Motion and Content for Video Generation

573 stars 113 forks source link

RuntimeError: arguments are located on different GPUs #30

Closed Fanny-Yuan closed 3 years ago

Fanny-Yuan commented 3 years ago

Hi, I was running your code on my GPUs but the error occurred. I tried to set one GPU but the problem still exists. I was wondering if you know how to solve that?

Fanny-Yuan commented 3 years ago

Supplement: I was running train.py using dynamic expression dataset.

Fanny-Yuan commented 3 years ago

cyhhhhhhit commented 3 years ago

Hi, I've got the same problem. Is there any new progress about this question?

sergeytulyakov commented 3 years ago

Hi, the framework was not implemented to support multigpu training. It has to be changed to allow for this. I'm happy to review a PR if anyone fixed this.