4

我正在尝试在 TensorFlow 中使用 tf.estimator 训练 alexnet 模型。训练过程很顺利，我可以看到日志显示得很好。

INFO:tensorflow:loss = 2.61362, step = 294
INFO:tensorflow:Saving checkpoints for 325 into /home/olu/Dev/data_base
sign_base/output/Checkpoints_N_Model/trained_alexnet_model/model.ckpt.
INFO:tensorflow:Loss for final step: 2.94104.

下面是如何调用训练函数：

traffic_sign_classifier.train(input_fn=train_input_fn,hooks=[logging_hook])

拜托，我怎样才能从 tf 估计器对象中将我的损失值作为普通的 python 浮点数

4

2 回答 2

3

完成训练后，您可以从 tensorboard 下载损失值。

在此处输入图像描述

于 2017-12-12T02:50:00.720 回答
1

估计器上的 .evaluate() 方法返回您可以在模型函数中指定的指标字典，在估计器规范中，https: //www.tensorflow.org/api_docs/python/tf/estimator/EstimatorSpec eval_metric_ops。我在 GitHub链接上的这个线程上找到了答案

于 2017-12-12T16:31:29.260 回答