tensorflow save_model模块

使用tensorflow save_model模块读取deepfm的W，V等参数，找了很多例子，都是读input，output,看到这篇讲的还算比较清楚：https://stackabuse.com/tensorflow-save-and-restore-models/?spm=a2c4e.11153940.blogcont228618.18.1dd47eaaQDV6Pd

具体实现：

import tensorflow as tf
from tensorflow import saved_model as smwith tf.Session() as sess:path = '/data/mayun/dfm/model_sum/serving_model_DEEP_FM/1'sm.loader.load(sess, tags=[sm.tag_constants.SERVING], export_dir=path)W = sess.run('deep_fm/fm_V:0')# vector = tf.nn.embedding_lookup(W, [146622390]) 报错：ValueError: Cannot create a tensor proto whose content is larger than 2GB.# print(sess.run(vector))# print(tf.nn.embedding_lookup(W, ['146622390']))print(W[146622390])

tensorflow save_model模块相关推荐

tensorflow contrib模块_OpenCV DNN 模块-风格迁移
本文主要介绍OpenCV的DNN模块的使用.OpenCV的DNN模块自从contrib仓库开始,就是只支持推理,不支持训练.但是仅仅只是推理方面,也够强大了.现在OpenCV已经支持TensorFlo ...
tensorflow: 图像处理模块 tf.image
tf.image.decode_jpeg tensorflow里面给出了一个函数用来读取图像,不过得到的结果是最原始的图像,是没有经过解码的图像,这个函数为tf.gfile.FastGFile('pa ...
anaconda安装的TensorFlow版本没有model这个模块
一.采用git bash来安装,确认已经安装了git 二.手动找到TensorFlow的模块文件夹地址,若不知道,输入以下两行代码: import tensorflow as tf tf.__path ...
TensorFlow Hub介绍：TensorFlow中可重用的机器学习模块库
摘要: 本文对TensorFlow Hub库的介绍,并举例说明其用法. 在软件开发中,最常见的失误就是容易忽视共享代码库,而库则能够使软件开发具有更高的效率.从某种意义上来说,它改变了编程的过程.我们 ...
深度学习框架 TensorFlow：张量、自动求导机制、tf.keras模块(Model、layers、losses、optimizer、metrics)、多层感知机(即多层全连接神经网络 MLP)
日萌社人工智能AI:Keras PyTorch MXNet TensorFlow PaddlePaddle 深度学习实战(不定时更新) 安装 TensorFlow2.CUDA10.cuDNN7.6. ...
简单粗暴上手TensorFlow 2.0，北大学霸力作，必须人手一册！
(图片付费下载自视觉中国) 整理 | 夕颜出品 | AI科技大本营(ID:rgznai100) [导读] TensorFlow 2.0 于近期正式发布后,立即受到学术界与科研界的广泛关注与好评.此前 ...
谷歌发布TensorFlow，用于测试人工智能模型的隐私保护
本文作者:郭仁贤 2020-06-25 17:36 导语:人工智能模型会泄露训练数据集的敏感信息,从而产生隐私风险. 近日,谷歌发布了隐私保护TensorFlow工具包,可以评估各种机器学习分类器的隐 ...
TensorFlow模型的签名推荐与快速上线\n
简介往期文章我们给你推荐一种TensorFlow模型格式介绍过, TensorFlow官方推荐SavedModel格式作为在线服务的模型文件格式.近期TensorFlow SavedModel模 ...
tensorflow2.0支持的python版本-TensorFlow 版本兼容性
本文面向需要在不同版本的 TensorFlow 之间向后兼容(针对代码或者数据)的用户,以及想要修改 TensorFlow 并同时保持兼容性的开发者. 语义化版本控制 2.0 TensorFlow 的 ...

tensorflow save_model模块

tensorflow save_model模块相关推荐

最新文章

热门文章