Closed FalconLK closed 5 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!
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
Thank you very much 👍
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!