CapsNet_Tensorflow icon indicating copy to clipboard operation
CapsNet_Tensorflow copied to clipboard

memory error

Open YifeiAI opened this issue 7 years ago • 0 comments

Hi,

I am using titan x (12G) to train the model on small_norb and it shows memory error when read in the data at : X = tf.convert_to_tensor(trainX, dtype=tf.float32) / 255.

Any ideas about why this happens? because i think 12G should be enouogh to load in all the data from small_norb dataset.

Thank you very much!

YifeiAI avatar Apr 29 '18 16:04 YifeiAI