search

meng-tang / rloss

Regularized Losses (rloss) for Weakly-supervised CNN Segmentation
MIT License
207 stars 46 forks source link

can I use multi-gpu for training ? #9

Open yaoqi-zd opened 5 years ago

yaoqi-zd commented 5 years ago

Hi, can I use multi-gpu for training ? since the densecrf loss layer is implemented in C++, I'm afraid that only single-gpu training is support ?