When you exercise, you should do sess.run only once. Recommend trying something like this, hope this helps:
with tf.Session() as sess: sess.run(tf.global_variables_initializer()) for i in range(train_set_size * epoch): train_step.run([optimizer, depth_loss, prediction])
Sergio g
source share