Yaoyi-Li / GCA-Matting

Official repository for Natural Image Matting via Guided Contextual Attention
MIT License
391 stars 62 forks source link

About the batch size is 10 for each GPU. #7

Open wty-ustc opened 4 years ago

wty-ustc commented 4 years ago

In you readme text, you write "Default training requires 4 GPUs with 11GB memory, and the batch size is 10 for each GPU. " But I think it‘s 10 for all GPU. May not right, so I consult you.

Yaoyi-Li commented 4 years ago

I think it is 10 for each GPU. Could you give me any hint that makes you think it's 10 for all GPU?