a68437552822520e8ca7c6a2c7a65ce4823a9e0c,finetune/base.py,BaseModel,_get_predict_input_fn,#BaseModel#Any#Any#,341

Before Change

        batch_size = batch_size or self.config.batch_size
        prefetch_buffer = 2  // breaks the pipeline to allow concurrency
        tf_dataset = self._dataset_without_targets(Xs)
        dataset = tf_dataset.batch(batch_size).prefetch(prefetch_buffer)
        return lambda: dataset.make_one_shot_iterator().get_next("Pred_dataset")

    def _inferrence(self, Xs, mode):
        estimator = self.get_estimator()

After Change

        batch_size = batch_size or self.config.batch_size
        prefetch_buffer = 2  // breaks the pipeline to allow concurrency
        tf_dataset = lambda: self._dataset_without_targets(Xs)
        return lambda: tf_dataset().batch(batch_size).prefetch(prefetch_buffer)

    def _inferrence(self, Xs, mode):
        estimator = self.get_estimator()
        input_func = self._get_predict_input_fn(Xs)

Instances