albertbup / deep-belief-network

A Python implementation of Deep Belief Networks built upon NumPy and TensorFlow with scikit-learn compatibility
MIT License
481 stars 212 forks source link

Run in GPU #52

Open YuchenKid opened 4 years ago

YuchenKid commented 4 years ago

For me, when I import from dbn.tensorflow, everything runs in CPU. And when I import from dbn, the pre-training part runs in GPU while the fine-tuning part still runs in CPU. I do not know why. Is there anything wrong with my environment settings?