f1fef359f319377ee0518a13ede38e4a654622ce,models_vggtiny.py,,model,#Any#Any#Any#Any#Any#Any#Any#,92

Before Change

    if data_format != "channels_last":
        // TODO: support NCHW
        print("data_format=%s is ignored" % data_format)

    with tf.variable_scope("model", reuse):
        //// Feature extraction part
        cnn = vgg_network(x)

After Change

        //// stage 2 ~ 6
        // for i in range(2, 7):
        with tf.variable_scope("stage1/branch2"):  // TODO: fix indent here and the names in npz
            for i in [5, 6]:  // only 3 stage in total
                b1, b2 = stage2(cnn, b1, b2, n_pos, mask_miss1, mask_miss2, is_train, scope_name="stage%d" % i)
                b1_list.append(b1)
                b2_list.append(b2)

        net = tl.layers.merge_networks([b1, b2])
        return cnn, b1_list, b2_list, net

Instances