search

affinelayer / pix2pix-tensorflow

Tensorflow port of Image-to-Image Translation with Conditional Adversarial Nets https://phillipi.github.io/pix2pix/
MIT License
5.07k stars 1.3k forks source link

Run on CUDA? #173

Open hjwdzh opened 5 years ago

hjwdzh commented 5 years ago

I notice that there is no explicit sentence to push tensors into gpu like with tf.device('/gpu:0'). Is it automatically run on GPU or do I need to do anything?