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.