carpedm20/ENAS-pytorch

Can you add an option to use or not use the GPU when calling PyTorch?

shyamalschandra opened this issue · 1 comments

shyamalschandra commented

Otherwise, I will do it myself and check it in! Thanks for your time!

gregoire22enpc commented

There is already this option. Just add --num_gpu 0 when running main.py in the console.

  • 👍1