TensorFlow
机器学习的首选
得食猫
这个作者很懒,什么都没留下…
展开
-
查看Tensorflow SavedModel模型信息
当我们拿到别人提供的pb模型时,我们很可能不清楚inputs所需的参数,可以使用以下方法来查看模型信息:找到你的tensorflow安装位置,依次定位到以下目录:tensorflow_core/python/tools执行如下命令:python saved_model_cli.py show --dir 模型路径 --all之后可以得到如下信息:signature_def['se...原创 2019-12-27 08:02:43 · 1268 阅读 · 0 评论 -
TensorFlow: CUDA_ERROR_OUT_OF_MEMORY
在运行别人已经训练好的模型时报了这个错TensorFlow: CUDA_ERROR_OUT_OF_MEMORY通过字面意思也知道是内存不够了,而CUDA指的是GPU。而我使用别人的模型,并不希望使用到GPU。为解决这个问题需要进行如下设置:import osos.environ["CUDA_DEVICE_ORDER"] = "PCI_BUS_ID"os.environ["CUDA_VI...原创 2019-12-19 08:01:45 · 417 阅读 · 0 评论 -
使用gRPC调用TensorFlow Serving后数据解析
我们都知道TensorFlow Serving支持RESTfull API和gRPC两种通讯方式,但使用gRPC通信后如何从响应中拿到我们所期望的值呢?,似乎很多人都遇到了这个问题,我也一样。stack overflow上已经有了答复,但并没能解决我的问题。先看下stack overflow上给的答复:采纳最多的:float_val = result.outputs['outputs'].f...原创 2019-12-17 07:09:38 · 764 阅读 · 1 评论