import tensorflow.contrib as contrib

            else:
                stitch1_1, stitch1_2 = fc1_1, fc1_2

            fc2_1 = contrib.layers.fully_connected(stitch1_1, 32, scope="fc2_1")
            fc2_2 = contrib.layers.fully_connected(stitch1_2, 32, scope="fc2_2")

            if cross_stitch_enabled:
                with tf.variable_scope("cross_stitch_2"):
