ad7d42093443bfdaeb1948e77f72b520419d39fc,magenta/models/image_stylization/image_stylization_train.py,,main,#Any#,66

Before Change

      total_loss, loss_dict = learning.total_loss(
          inputs, stylized_inputs, style_gram_matrices, content_weights,
          style_weights)
      for key, value in loss_dict.iteritems():
        tf.summary.scalar(key, value)

      // Set up training

After Change

        tf.summary.scalar(key, loss_dict[key])

      // Adding Image summaries to the tensorboard.
      tf.summary.image("image/0_inputs", inputs, 3)
      tf.summary.image("image/1_styles", style_images, 3)
      tf.summary.image("image/2_styled_inputs", stylized_inputs, 3)

      // Set up training
      optimizer = tf.train.AdamOptimizer(FLAGS.learning_rate)
      train_op = slim.learning.create_train_op(

Instances