marcbelmont / cnn-watermark-removal

Fully convolutional deep neural network to remove transparent overlays from images
1.24k stars 228 forks source link

Validation step seems to have infinite loop #31

Open kmrabhay opened 5 years ago

kmrabhay commented 5 years ago

Hi,

            sess.run(iterator_inits[1])  # switch to validation dataset
            while True:
                import pdb; pdb.set_trace();
                try:
                    _, summaries_ = sess.run([loss, summaries],
                                             feed_dict={training: False})
                    # print(summaries_)

                    val_writer.add_summary(summaries_, global_step_)

                except tf.errors.OutOfRangeError:
                    break

Above code for validation steps never finishes. I trained on 192 images and validation steps never finishes even after 10 hours.

Is the because of that infinite loop and never reaching to break statement?