ensorflow.python.framework.errors_impl.InvalidArgumentError: Reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero..'

[cchantra]

I got this error when I retrain the network with my data set.

'tensorflow.python.framework.errors_impl.InvalidArgumentError: Reshape cannot infer the missing input size for an empty tensor unless all specified input sizes are non-zero..'

I use a small data set of 525 and class_num is 6. How to modify the number of iterations and the 50,0000 value in the code? (In SE_Inception_v4.py)

I already put the new iteration value to a smaller value, say (iteration_n = len(train_x)/batch_size+ 1)
where batch_sze=128 and use this:

if pre_index + batch_size < batch_size*iteration_n: #in place of 50000
....
Any help is appreciated.

[ahmad53135]

it seems it is because of low gpu memory