![](https://img-blog.csdnimg.cn/20201014180756927.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
Tensorflow
开心就哈哈
这个作者很懒,什么都没留下…
展开
-
tensorflow InternalError (see above for traceback): Blas SGEMM launch failed : m=1228800, n=15, k=30
环境: 16.04.1-Ubuntu python 3.6.10 tensorflow-gpu 1.12.0 CUDA Version 9.0.176 Cudnn 7.6.4 问题: 运行代码出现:InternalError (see above for traceback): Blas SGEMM launch failed : m=1228800, n=15, k=30 错误。...原创 2020-03-10 18:21:56 · 1106 阅读 · 0 评论 -
使用tensorflow 提取特征图(How to use tensorflow to extract feature map from checkpoint)
如何使用tensorflow 提取特征图 ? 》首先模型训练结束后,我们会得到关于【3】【4】检查点的四个文件:.data, .index, .meta, checkpoint; 其中.meta是图结构,也就是神经网络的结构,在训练过程中图结构水不不变的,保存一次 即可。实现:saver = tf.train.Saver(), saver.save(less, ‘model-name’,w...原创 2020-03-02 20:57:08 · 1679 阅读 · 0 评论 -
基于tensorflow实现,导入预训练模型之后,将global_step, epoch重新设置为0
问题:使用tensoflow导入训练模型参数后,eoch, global_step仍然是之前训练所保留的值。因此,在导入预训练模型参数之后,需要将epoch, global_step的值重新置0. 部分参考代码: with tf.Session(config=config) as sess: # Initialize all variables ...原创 2019-10-23 15:49:19 · 1483 阅读 · 4 评论