Skip to content

GitLab

Commit 14797cf7 authored by Joseph Friedman's avatar Joseph Friedman
Browse files

updated eval for python 3

parent fad60753
Hide whitespace changes
Inline Side-by-side
Showing with 2 additions and 2 deletions
research/im2txt/im2txt/evaluate.py
View file @ 14797cf7
...@@ -76,7 +76,7 @@ def evaluate_model(sess, model, global_step, summary_writer, summary_op): ...@@ -76,7 +76,7 @@ def evaluate_model(sess, model, global_step, summary_writer, summary_op):
start_time = time.time() start_time = time.time()
sum_losses = 0. sum_losses = 0.
sum_weights = 0. sum_weights = 0.
for i in xrange(num_eval_batches): for i in range(num_eval_batches):
cross_entropy_losses, weights = sess.run([ cross_entropy_losses, weights = sess.run([
model.target_cross_entropy_losses, model.target_cross_entropy_losses,
model.target_cross_entropy_loss_weights model.target_cross_entropy_loss_weights
...@@ -143,7 +143,7 @@ def run_once(model, saver, summary_writer, summary_op): ...@@ -143,7 +143,7 @@ def run_once(model, saver, summary_writer, summary_op):
global_step=global_step, global_step=global_step,
summary_writer=summary_writer, summary_writer=summary_writer,
summary_op=summary_op) summary_op=summary_op)
except Exception, e: # pylint: disable=broad-except except Exception as e: # pylint: disable=broad-except
tf.logging.error("Evaluation failed.") tf.logging.error("Evaluation failed.")
coord.request_stop(e) coord.request_stop(e)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment