if has_att_heads:
        attribute_heads = [
            dict(
                name='depth',
                type='regression',