CODEWITHSUNDEEP
tensorflowconv-neural-networktensorflow-estimator
Naz_Jnr
Naz_Jnr

Reputation: 580

TensorFlow - How to Get My Loss Value from tf.Estimaor

I am trying to train an alexnet model using tf.estimator in TensorFlow. The training process works smoothly, and I can see the logs well displayed.

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.

Below is how the training function is called:

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

Please, how can I get my loss value as a normal python floating point from the tf estimator object

Upvotes: 4

Views: 2698

Answers (2)

Naz_Jnr
Naz_Jnr

Reputation: 580

The .evaluate() method on the estimator returns a dictionary of metrics you can specify in your model function, in the estimator spec, https://www.tensorflow.org/api_docs/python/tf/estimator/EstimatorSpec eval_metric_ops. I found the answer on this thread on GitHub Link

Upvotes: 1

secsilm
secsilm

Reputation: 430

You can download the loss values from tensorboard when finishing your training.

enter image description here

Upvotes: 3

Related Questions