机器学习
for_ge
这个作者很懒,什么都没留下…
展开
-
tensorflow keras重复运行结果不一致问题
tensorflow keras重复运行结果不一致问题 tensorflow版本为:2.2.0,在使用tensorflow中的keras构建深度学习网络时,每次的训练结果都不一致。添加 shuffle=False后,没有效果,最后使用 tf.random.set_seed(1)函数,经测试,得到的训练结果固定了。 使用tf.random.set_seed()时,最好放在导入tensorflow后,否则训练结果无法固定 ...转载 2020-12-20 17:03:28 · 2597 阅读 · 1 评论 -
TensorFlow-cifar10-图像分类之网络结构
目录 LeNet网络: AlexNet网络: VGG16网络: VGG19网络: LeNet网络: def LeNet(inputs): mu = 0 sigma = 0.1 print(inputs.shape) # TODO: 第一层卷积:输入=32x32x3, 输出=28x28x6 conv1_w = tf.Variable(tf.t...转载 2020-03-06 17:53:31 · 789 阅读 · 0 评论 -
保存模型的参数Tensorflow/Keras
Tensorflow: 保存模型: saver = tf.train.Saver() saver.save(sess, "{}.ckpt") 加载模型: saver = tf.train.import_meta_graph('{}.ckpt.meta') saver.restore(sess, {}.ckpt) 修改模型某些参数 with tf.Session(grap...原创 2020-03-06 17:56:03 · 872 阅读 · 0 评论