search

crestonbunch / tbcnn

Efficient tree-based convolutional neural networks in TensorFlow
MIT License
150 stars 44 forks source link

Regarding the TBCNN Impl. #9

Closed FalconLK closed 5 years ago

FalconLK commented 6 years ago

Hello,

Thank you for the great impl.

I've been trying to understand your impl. By the way, does the impl also work without CUDA? if not, which part of code should I modify to make it work without CUDA?

Thank you very much and have a great day!

bdqnghi commented 5 years ago

this implementation is based on tensorflow, you can enable Cuda training by setting something like :

os.environ['CUDA_VISIBLE_DEVICES'] = "1"

to enable GPU training

FalconLK commented 5 years ago

Thank you very much 👍