dd3c13d51975d7ca569681dd19639473d19e8e85,server/bert_serving/server/__init__.py,BertWorker,run,#BertWorker#,329

Before Change

    def run(self):
        print("________")
        os.environ["CUDA_VISIBLE_DEVICES"] = "6"

        import tensorflow as tf
        from tensorflow.python.client import device_lib

After Change

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

        context = zmq.Context()
        receiver = context.socket(zmq.PULL)
        receiver.connect(self.worker_address)

        sink = context.socket(zmq.PUSH)
        sink.connect(self.sink_address)

Instances