610718c4e5072a98ccb79fbca4b58febb5c6267f,magenta/models/image_stylization/image_stylization_train.py,,main,#Any#,67

Before Change

          summarize_gradients=False)

      // Function to restore VGG16 parameters.
      init_fn_vgg = slim.assign_from_checkpoint_fn(vgg.checkpoint_file(),
                                                   slim.get_variables("vgg_16"))

      // Run training
      slim.learning.train(
          train_op=train_op,

After 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

Instances