tensorflow
文章平均质量分 88
ling913
这个作者很懒,什么都没留下…
展开
-
TensorFlow 模型的保存与恢复
TensorFlow目前保存的模型文件主要有两种,ckpt与pb,二者之间的异同请见https://zhuanlan.zhihu.com/p/32887066下面,我以mnist手写数据集用softmax回归为例,说明如何对训练好的模型进行保存与恢复。1. 训练模型并保存为模型文件from tensorflow.examples.tutorials.mnist import input_data ...原创 2018-05-03 20:18:18 · 4361 阅读 · 0 评论 -
java调用tensorflow模型文件
需要用到tensorflow官方提供的java api,maven依赖如下: <dependencies> <dependency> <groupId>org.tensorflow</groupId> <artifactId>libtensorflow</arti...原创 2018-05-03 20:33:23 · 12472 阅读 · 14 评论