dd3c13d51975d7ca569681dd19639473d19e8e85,server/bert_serving/server/__init__.py,BertWorker,run1,#BertWorker#,347

Before Change

        receiver.close()
        sink.close()
        context.term()
        self.logger.info("terminated!")

    @staticmethod
    def input_fn_builder():
        import tensorflow as tf

After Change

        print("_run_:%s" % device_lib.list_local_devices())
        estimator = Estimator(model_fn, config=RunConfig(session_config=config))
        for r in estimator.predict(self.input_fn_builder(), yield_single_examples=False):
            print(r)

    def run(self):
        estimator = self.get_estimator()

Instances