Open bodokaiser opened 7 years ago
Use with tf.device('/cpu:0') for pre processing and with tf.device('/gpu:0') for training and inference.
with tf.device('/cpu:0')
with tf.device('/gpu:0')
Use
with tf.device('/cpu:0')
for pre processing andwith tf.device('/gpu:0')
for training and inference.