摘要：
import uuid identity = str(uuid.uuid4()).encode('ascii') 阅读全文
摘要：
class TNLog(object): def __init__(self, log_dir, context, verbose): if not os.path.exists(log_dir): os.mkdir(log_dir) self.context = context self.root_log_dir = log... 阅读全文
摘要：
当我们把训练好的tensorflow训练图拿来进行预测时，会有多个训练时生成的节点，这些节点是不必要的，我们需要在预测的时候进行删除。 下面以bert的图为例，进行优化 返回一个gfile类型的文件，我们可以像原来导入模型文件时，恢复图，不过这个图是优化过的。 阅读全文
摘要：
因为涉及到同一台电脑多个GPU，在指定tensorflow图的时候，需要为不同的图指定不同的GPU，所以必须在导入tensorflow之前，指定可用的GPU 阅读全文
摘要：
saver = self.tf_instance.train.Saver() self.sess = self.tf_instance.Session(config=sess_config, graph=graph) self.sess.run(self.tf_instance.global_variables_initializer()) # 在restore时加入这行代码 self.tf_i... 阅读全文
摘要：
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_we... 阅读全文
摘要：
1.先获取节点 2.获取图 阅读全文