ckpt_file = os.path.join(self.args.bert_dir, 'model.ckpt-6123') # 獲取圖的reader reader = self.tf_instance.train.NewCheckpointReader(ckpt_file) # 根據節點名稱獲取節點 output_weights = reader.get_tensor('output_weights') output_bias = reader.get_tensor('output_bias')