how to train on gpu - Githubissues

hyperconnect / MMNet

Code for Towards Real-Time Automatic Portrait Matting on Mobile Devices

https://arxiv.org/abs/1904.03816v1

Apache License 2.0

177 stars 36 forks source link

Open sunzhe09 opened 5 years ago

sunzhe09 commented 5 years ago

I found it train on CPU

shurain commented 5 years ago

If you have GPU enabled, it should exploit GPUs.