def classification_head_generator(params): """Generator function for classification head architecture.""" return heads.ClassificationHead( params.num_classes, params.endpoints_num_filters, params.aggregation, params.dropout_rate, batch_norm_relu=batch_norm_relu_generator(params.batch_norm))
def classification_head_generator(params): """Generator function for classification head architecture.""" head_params = params.classification_head return heads.ClassificationHead( params.architecture.num_classes, head_params.endpoints_num_filters, head_params.aggregation, head_params.dropout_rate, batch_norm_activation=batch_norm_activation_generator( params.batch_norm_activation))