Skip to content
This repository has been archived by the owner on Jul 7, 2023. It is now read-only.

Commit

Permalink
Merge pull request #210 from ajbouh/ajbouh-python3
Browse files Browse the repository at this point in the history
Fix Python3 support in model_builder.py
  • Loading branch information
lukaszkaiser authored Aug 7, 2017
2 parents 2758f5b + 51279cf commit c5e13db
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion tensor2tensor/utils/model_builder.py
Original file line number Diff line number Diff line change
Expand Up @@ -195,7 +195,7 @@ def nth_model(n):
features, skip=(skipping_is_on and skip_this_one))
with tf.variable_scope("losses_avg", reuse=True):
total_loss, ops = 0.0, []
for loss_key, loss_value in losses_dict.iteritems():
for loss_key, loss_value in six.iteritems(losses_dict):
loss_moving_avg = tf.get_variable("problem_%d/%s_loss" % (n,
loss_key))
ops.append(
Expand Down

0 comments on commit c5e13db

Please sign in to comment.