Commit b6ebdd69 authored by cclauss's avatar cclauss Committed by GitHub
Browse files

from six.moves import range for Python 3

@cshallue
parent a03b3638
...@@ -22,6 +22,8 @@ from __future__ import absolute_import ...@@ -22,6 +22,8 @@ from __future__ import absolute_import
from __future__ import division from __future__ import division
from __future__ import print_function from __future__ import print_function
from six.moves import range
import math import math
import os.path import os.path
import time import time
...@@ -75,7 +77,7 @@ def evaluate_model(sess, losses, weights, num_batches, global_step, ...@@ -75,7 +77,7 @@ def evaluate_model(sess, losses, weights, num_batches, global_step,
start_time = time.time() start_time = time.time()
sum_losses = 0.0 sum_losses = 0.0
sum_weights = 0.0 sum_weights = 0.0
for i in xrange(num_batches): for i in range(num_batches):
batch_losses, batch_weights = sess.run([losses, weights]) batch_losses, batch_weights = sess.run([losses, weights])
sum_losses += np.sum(batch_losses * batch_weights) sum_losses += np.sum(batch_losses * batch_weights)
sum_weights += np.sum(batch_weights) sum_weights += np.sum(batch_weights)
......
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