Just a little question about multi-gpu model

xmengli / H-DenseUNet

TMI 2018. H-DenseUNet: Hybrid Densely Connected UNet for Liver and Tumor Segmentation from CT Volumes

526 stars 161 forks source link

Just a little question about multi-gpu model #35

Closed aa1234241 closed 5 years ago

aa1234241 commented 5 years ago

Hello, in your code, you write your own mult-gpu model. I am a little confused since keras has native multi-gpu support. Is there any problem to use the parallel model in original Keras?

Sincerely

leetesua commented 5 years ago

same here

xmengli commented 5 years ago

You can use the Keras in this code. It contains the multi-gpu function.

aa1234241 commented 5 years ago

Thank you very much.