search

hellozhuo / pidinet

Code for the ICCV 2021 paper "Pixel Difference Networks for Efficient Edge Detection" (Oral).
Other
445 stars 69 forks source link

请问如何多卡运行？ #18

Closed lian666-ch closed 2 years ago

zhuoinoulu commented 2 years ago

你好，这个需要修改main.py，目前我们还没有具体实现。多卡运行应该需要输入图片size一致(这样可以使得batch size > 1从而均分给多个GPU). 所以实现方案可以是先对输入做一个预处理，使得size一致，然后可以调用DistributedDataParallel完成多GPU运行。

当然并不排除size不一致也能多卡运行的可能 ఠ_ఠ，这种情况可能再需要调研一下~

lian666-ch commented 2 years ago

谢谢