tensorflow
奋斗的瘦胖子
作为一个码农小渣渣,开启执着的自驱学习之路!
展开
-
解决TensorFlow只能保存几个模型的问题
应该有很多人和我一样,不是系统的学习tensflow,然后用别人写的代码;直接调整下就训练;然后遇到模型保存总是有消失的情况;模型保存有一个max_to_keep参数;如果设置了固定数字;当模型数量超过这个数字之后,tensorflow就会清理历史模型;只保留最新的;所以大家使用过程中可以按需调整saver = tf.train.Saver(max_to_keep=10)...原创 2021-03-09 09:04:13 · 355 阅读 · 0 评论 -
Java加载tensorflow模型(单机加载或分布式加载)
我们常用python进行tensorflow深度模型训练,然后,训练后的模型需要应用到web端调用,或者app应用调用,甚至分布式任务使用等等。在这些应用中java代码调用,是避免不了的。本文就介绍一下,java加载tensorflow模型的方式,分别单机调用和分布式调用。1.环境需要首先需要加载一些jar包,如果是maven项目在pom.xml中添加以下依赖。至于版本信息,切记一定要和你py...原创 2019-10-31 13:59:22 · 2566 阅读 · 0 评论 -
TensorFlow模型的保存及模型的应用
tensorflow的模型保存形式?1.ckpt格式就是通过如下几个函数实现的tensorflow模型保存的模型,是ckpt格式的模型。saver = tf.train.Saver()...saver.save(sess, saveFile)就可以保存出如下文件:checkpointmodel-450.data-00000-of-00001model-450.indexmod...原创 2019-10-31 11:16:02 · 457 阅读 · 0 评论