def conv5_sep(inputs): with tf.variable_scope(get_layer_id()): with tf.variable_scope('branch_3'): out = conv_op( inputs[0][0], 5, is_training, out_filters, out_filters, self.data_format, start_idx=None, separable=True) out = post_process_out(out, inputs[1]) return out
def conv3(inputs): # res_layers is pre_layers that are chosen to form skip connection # layers[-1] is always the latest input with tf.variable_scope(get_layer_id()): with tf.variable_scope('branch_0'): out = conv_op( inputs[0][0], 3, is_training, out_filters, out_filters, self.data_format, start_idx=None) out = post_process_out(out, inputs[1]) return out