当我们把训练好的tensorflow训练图拿来进行预测时,会有多个训练时生成的节点,这些节点是不必要的,我们需要在预测的时候进行删除. 下面以bert的图为例,进行优化 def optimize_graph(self, checkpoint_file, model_config): import json tf = self.import_tf() from tensorflow.python.tools.optimize_for_inference_lib import optimize_fo…