What should I do if I use Your Excellent Work in CPU mode?

wentaozhu / DeepLung

WACV18 paper "DeepLung: Deep 3D Dual Path Nets for Automated Pulmonary Nodule Detection and Classification"

Apache License 2.0

315 stars 143 forks source link

What should I do if I use Your Excellent Work in CPU mode? #12

Closed TracelessLe closed 6 years ago

TracelessLe commented 6 years ago

Recently, I'm trying to use part of your excellent work in nodule dectection. Be more exact, I use the detection net of Deeplung to test on my CT data. But I need to transform the GPU mode into CPU, what should I modify in the test part of "main.py" in the "detector" folder? I hope to receive your reply. Thanks.

wentaozhu commented 6 years ago

You can take a look at how to use CPU in pytorch. We use pytorch for the implementation. Thank you very much for your interest!

TracelessLe commented 6 years ago

Thanks a lot.