
tensorflow
武小胖儿
这个作者很懒,什么都没留下…
展开
-
模型推理过程中tensorboard的profiler的使用方法
一、tensorboard的简介https://www.tensorflow.org/guide/profilertensorboard是tensorflow集成的一个可视化工具,用于展示模型运行过程中的一些特征,包括运行时间、中间数据的图形化展示等等(详细看官方解释)。本文主要介绍tensorflow profiler的使用方法。二、安装profiler的安装说明1. 保证已经成功安装了一下内容TensorFlow >= 2.2.0 TensorBoard >= 2.原创 2020-12-04 10:19:44 · 2560 阅读 · 1 评论 -
tensorflow 显存空间不足
tensorflow在显存管理方面有缺陷,只能使用权宜之计https://github.com/tensorflow/tensorflow/issues/36465#issuecomment-582749350https://github.com/tensorflow/tensorflow/issues/36465原创 2020-11-30 16:24:33 · 447 阅读 · 0 评论 -
tensorflow V2.2的学习笔记及2.2和1.14的对比
教是学最好的方式,fightingtensorflow V2.2的安装tensorflow常见的安装途径包括两个:pip安装和conda安装(强烈推荐后者,后者具备如下安装优势:1. 速度快,2. 自动安装与tensorflow版本配套的cuda和cudnn【手动安装的时候一定要参考以下链接,找到对应的版本https://www.tensorflow.org/install/source#linux并采用正确的方式安装https://blog.youkuaiyun.com/panligo/article/det原创 2020-11-23 21:39:35 · 901 阅读 · 1 评论 -
tensorflow keras权重加载过程的源码解析
整体加载过程梳理1. 读取h5文件中layer_name的内容,并获取layer_name下的weight信息,按先layer_name后weight_name的形式把所有weight存放在数组里。2. 将图中的layers按照key=layer_name, value=layer的方式重新组织所有内容,并将结果存放到dict里。(多个layer可能公用一个layer_name,这些重名的layers被组织到同一个数组下)3. 遍历h5下的所有layer,读取特定layer下的所有weight,原创 2020-11-23 09:58:38 · 1853 阅读 · 3 评论 -
cuda安装
https://blog.youkuaiyun.com/panligo/article/details/86627289转载 2020-11-17 08:45:59 · 126 阅读 · 0 评论 -
测试tensorflow alexnet
alexnet源码:https://github.com/kratzert/finetune_alexnet_with_tensorflow上述代码无运行错误,但是无法给出正确的识别结果,解决方案如下:import osimport cv2import numpy as npimport tensorflow as tfimport matplotlib.pyplot as ...原创 2019-05-22 18:46:42 · 207 阅读 · 0 评论 -
tensorflow设置layer的执行位置(device)
1.tensoflow默认的执行位置https://tensorflow.google.cn/guide/using_gpu1.1 设备选择的优先级If a TensorFlow operation has both CPU and GPU implementations, the GPU devices will be given priority when the operatio...原创 2019-06-03 17:05:51 · 1487 阅读 · 2 评论