![](https://img-blog.csdnimg.cn/20201014180756919.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
tensorflow
Sesen_s
这个作者很懒,什么都没留下…
展开
-
tensorflow查看模型checkpoint保存节点
from tensorflow.python import pywrap_tensorflowimport oscheckpoint_path=os.path.join('model.ckpt-131805')reader=pywrap_tensorflow.NewCheckpointReader(checkpoint_path)var_to_shape_map=reader.get_...原创 2019-12-13 12:24:18 · 616 阅读 · 0 评论 -
TensorFlow实现模型断点训练,checkpoint模型载入
深度学习中,模型训练一般都需要很长的时间,由于很多原因,导致模型中断训练,下面介绍继续断点训练的方法。方法一:载入模型时,不必指定迭代次数,一般默认最新# 保存模型saver = tf.train.Saver(max_to_keep=1) # 最多保留最新的模型# 开启会话with tf.Session() as sess: # saver.restore(sess, '...原创 2018-08-04 13:51:55 · 17148 阅读 · 7 评论 -
Tensorflow(二)单机多卡分布式训练
建立分布式训练:# 计算losses:with tf.device('/gpu:0'): D_real_loss = tf.reduce_mean(tf.nn.sigmoid_cross_entropy_with_logits(logits=D_logit_real, ...原创 2018-08-04 19:26:33 · 3705 阅读 · 2 评论 -
TensorBoard的一波三折:如何用笔记本远程访问服务器,查看tensorboard曲线
摒弃之前的隧道做法,以下两种简单粗暴1、将服务器上训练好的模型log,拷贝到本地电脑 运行tensorboard --logdir = log 然后直接访问生成地址2、服务器上运行命令:tensorboard --logdir = log --port=8008(log路径,因人而异;端口号因人而异) 生成地址:http://node01:8008;其...原创 2018-11-23 10:44:38 · 3222 阅读 · 2 评论 -
tensorflow summary错误:tags and values not the same shape [ ] != [16]
训练模型时,将loss值保存进log,随后使用tensorboard --logdir=log,查看loss曲线。出现tags and values not the same shape这个错误的原因,是因为对tf.summary.scalar()函数没有理解。解决办法: tf.summary.scalar(loss.op.name,loss)有两个参数。第一个参数表示...原创 2018-12-08 20:26:15 · 5230 阅读 · 2 评论 -
远程访问服务器,ImportError: libcusolver.so.9.0: cannot open shared object file: No such file or directory问题
在远程访问服务器时,在import tensorflow时,经常遇到ImportError: libcusolver.so.9.0: cannot open shared object file: No such file or directory问题,先将解决方法记录下来,方便以后使用。1、root用户下,在/etc/profile中配置全局变量命令为:############...原创 2018-12-25 17:18:07 · 1708 阅读 · 0 评论