tensorflow - 如何构建具有独立推理和训练部分的 TF Graph?

标签 tensorflow

引用 this post之前问过,因为建议创建一个具有独立推理和训练部分的图。

样板代码将不胜感激。

最佳答案

存储库中的 MNIST 卷积就是一个例子——tensorflow/tensorflow/models/image/mnist/convolutional.py

当您将模型构建代码分解为函数( model in convolutional.py )时,它遵循一种模式,并为 eval 和 training 部分分别调用它

 logits = model(train_data_node, True)
 loss = tf.reduce_mean(tf.nn.sparse_softmax_cross_entropy_with_logits(
      logits, train_labels_node))
 eval_prediction = tf.nn.softmax(model(eval_data))

对于培训,您可以输入 train_data_node并最小化 loss ,对于 eval,你输入 eval_data节点并在 eval_prediction 处获得结果

关于tensorflow - 如何构建具有独立推理和训练部分的 TF Graph?,我们在Stack Overflow上找到一个类似的问题: https://stackoverflow.com/questions/40352841/

相关文章:

python - 如何在带有 hub.KerasLayer 的 Tensorflow 2.0 中使用自动混合精度

python - 如何使用 tensorflow 注意层?

TensorFlow:恢复 RNN 网络后损失猛增

docker - 带有 TensorFlow 后端的 Keras 不使用 GPU

python - tensorflow : ValueError: Shape must be rank 2 but is rank 3

machine-learning - Tensorflow:权重衰减与 Logits 标准化

tensorflow - 如何在 Keras 中构建这个自定义层?

python - 计算 tensorflow 梯度

python - Tensorflow 线性回归结果与 Numpy/SciKit-Learn 不匹配

python - 如何使用功能 API 训练 Keras 模型,该模型具有两个输入和两个输出,并使用两个 ImageDataGenerator 方法 (flow_from_directory)