There is a bug when I try to train with Multi-GPU.

richzhang / PerceptualSimilarity

LPIPS metric. pip install lpips

https://richzhang.github.io/PerceptualSimilarity

BSD 2-Clause "Simplified" License

3.64k stars 499 forks source link

There is a bug when I try to train with Multi-GPU. #35

Closed gsygsy96 closed 5 years ago

gsygsy96 commented 5 years ago

I guess there is a bug when you put your model to mutil-gpus.

Reference: https://github.com/pytorch/pytorch/issues/8637#issuecomment-398268602

gsygsy96 commented 5 years ago

To solve it, using nn.ModuleList() to warp self.lins in model/network_basics.py.

LucaSoato commented 4 years ago

Please, can you tell me a little bit more about your solution? 😃 Sadly I'm stuck there while trying to train on multiple gpus for camvid segmentation.