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

Before Change

      // Function to restore VGG16 parameters
      // TODO(iansimon): This is ugly, but assign_from_checkpoint_fn doesn"t
      // exist yet.
      saver = tf.train.Saver(slim.get_variables("vgg_16"))
      def init_fn(session):
        saver.restore(session, vgg.checkpoint_file())

      // Run 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