Problem with running on GPU

mesutpiskin / id-card-detector

:credit_card: Detecting the National Identification Cards with Deep Learning (Faster R-CNN)

MIT License

283 stars 120 forks source link

Closed ramintoosi closed 5 years ago

ramintoosi commented 5 years ago

Hi, could you please tell me how can i run your code on a GPU? I'm not very familiar with tensorflow i usually use pytorch.