search

shunsukesaito / PIFu

This repository contains the code for the paper "PIFu: Pixel-Aligned Implicit Function for High-Resolution Clothed Human Digitization"
https://shunsukesaito.github.io/PIFu/
Other
1.75k stars 342 forks source link

DataParallel not called for model init #94

Closed RohanChacko closed 3 years ago

RohanChacko commented 3 years ago

Hi,

I noticed that HGPIFuNet isn't actually using multi-gpus. The model returned from init_net fn is not returned to anything when called here. Same is the case for ResBlkPIFuNet here. The initialized model runs only one gpu as shown here.

Please let me know if I'm missing out on something.

shunsukesaito commented 3 years ago

Yes, the current code base doesn't support muliti-GPUs and I don't plan to add it myself at this moment (pull request is always welcome though.)