search

poppinace / indexnet_matting

(ICCV'19) Indices Matter: Learning to Index for Deep Image Matting
Other
389 stars 90 forks source link

train on multi gpu #24

Closed QWERDF007 closed 4 years ago

QWERDF007 commented 4 years ago

How to train on multi gpu? thanks

poppinace commented 4 years ago

Hi, the code is compatible on multi GPU training. Just set CUDA_VISIBLE_DEVICES=0,1,2,3 if you have 4 GPUs and set sync_bn = True.

QWERDF007 commented 4 years ago

thanks!