f9a33006e5794fe1bfae5be04c44a58c68f6696f,magenta/models/image_stylization/image_stylization_finetune.py,,main,#Any#,72

Before Change

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

      // Function to restore N-styles parameters.
      // TODO(iansimon): This is ugly, but assign_from_checkpoint_fn doesn"t
      // exist yet.
      saver_n_styles = tf.train.Saver(other_vars)
      def init_fn_n_styles(session):
        saver_n_styles.restore(session, os.path.expanduser(FLAGS.checkpoint))

      def init_fn(session):

After Change

                    if "InstanceNorm" not in var.name]

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


      // Function to restore N-styles parameters.
      init_fn_n_styles = slim.assign_from_checkpoint_fn(os.path.expanduser(FLAGS.checkpoint),
                                                        other_vars)

      def init_fn(session):
        init_fn_vgg(session)
        init_fn_n_styles(session)

Instances