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

Before Change

            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":
                self.sess.run(self.iterator.initializer, feed_dict=self.iterator_arg)
                return self.sess.run(next_batch)
            elif self.iterator_type == "reinitializable":
                self.sess.run(self.iterator_arg)
                return self.sess.run(next_batch)
            else:
                self.sess.run(self.iterator_arg[0].initializer)
                return self.sess.run(next_batch, feed_dict=self.iterator_arg[1])

Instances