
tensorflow
文章平均质量分 68
Panda_Boy_yi
华科计算机专业研究生,做图形学
展开
-
tensorflow 中dataset模块迭代器的用法
本文主要讲述tensorflow中迭代器的用法import tensorflow as tf dateset1 = tf.data.Dataset.from_tensor_slices(tf.random_uniform([4, 10], 0, 8)) print(dateset1.output_types) #"tf.float32" print(dateset1.output_shapes)...原创 2018-03-08 10:50:04 · 2482 阅读 · 0 评论 -
tensorflow 模型与数据的存储与恢复
我们在tensorflow中训练模型时,往往一个模型需要训练好几天,训练中可能出现意外关机而停止训练。如果此时再重头开始,必然使得前面已经训练的结果功亏一篑。于是,就有了save与restore。变量在文件中以name为名称存储import tensorflow as tf # 定义存储路径 save_path = "./test/model.ckpt" # 存储变量 v1 = tf.get_v...原创 2018-03-09 15:41:53 · 735 阅读 · 0 评论