3be6b6d054feb8cfdd6fdeee1df6d138e193aa3d,art/data_generators.py,TFDataGenerator,get_batch,#TFDataGenerator#,200

Before Change

        import tensorflow as tf

        if isinstance(self.iterator, tf.data.Iterator):
            try:
                return self.sess.run(self.iterator.get_next())
            except tf.errors.OutOfRangeError as err:
                raise ValueError("Could not extract next batch. " + str(err))

After Change

            if self.iterator_type == "initializable":
                return self.sess.run(next_batch)
            elif self.iterator_type == "reinitializable":
                return self.sess.run(next_batch)
            else:
                return self.sess.run(next_batch, feed_dict=self.iterator_arg[1])
        except (tf.errors.FailedPreconditionError, tf.errors.OutOfRangeError):
            if self.iterator_type == "initializable":

Instances