4ec258a2261a943910a34be1b7d53a646f089367,luminoth/train.py,,run,#Any#Any#Any#Any#Any#Any#Any#Any#Any#,19

Before Change

            trace_level=tf.RunOptions.FULL_TRACE
        )

    if is_chief:
        // Load pretrained weights needs to be called before defining the train
        // op. After it, variables for the optimizer are created.
        with tf.control_dependencies([tf.global_variables_initializer()]):
            with tf.control_dependencies([model.load_pretrained_weights()]):
                init_op = tf.no_op(name="global_init_load_pretrained")
    else:
        init_op = tf.no_op()

    // Create custom Scaffold to make sure we run our own init_op when model
    // is not restored from checkpoint.
    summary_op = [model.summary]
    summaries = tf.summary.merge_all()
    if summaries is not None:
        summary_op.append(summaries)

After Change

    scaffold = tf.train.Scaffold(
        // Initialize global variables.
        init_op=tf.global_variables_initializer() if is_chief else tf.no_op(),
        // Queue-related variables need a special initializer.
        local_init_op=tf.local_variables_initializer(),
        summary_op=summary_op,

Instances