[
  {
    "metadataOutputVersion" : "3.0",
    "storagePrecision" : "Mixed (Float16, Palettized (4 bits))",
    "outputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1 × 6)",
        "shortDescription" : "",
        "shape" : "[1, 6]",
        "name" : "probs",
        "type" : "MultiArray"
      }
    ],
    "modelParameters" : [

    ],
    "specificationVersion" : 7,
    "mlProgramOperationTypeHistogram" : {
      "Transpose" : 11,
      "Ios16.maxPool" : 4,
      "Squeeze" : 1,
      "Ios16.softmax" : 3,
      "Ios16.linear" : 10,
      "Ios16.add" : 7,
      "Ios16.realDiv" : 1,
      "Ios16.reduceSum" : 2,
      "ExpandDims" : 4,
      "Ios16.sub" : 1,
      "Ios16.clip" : 1,
      "Ios16.cast" : 3,
      "Ios16.constexprLutToDense" : 12,
      "Ios16.relu" : 4,
      "Ios16.layerNorm" : 4,
      "SliceByIndex" : 6,
      "Ios16.conv" : 3,
      "Ios16.matmul" : 4,
      "Ios16.gelu" : 2,
      "Ios16.reshape" : 8,
      "Ios16.mul" : 5
    },
    "computePrecision" : "Mixed (Float16, Float32, Int32)",
    "isUpdatable" : "0",
    "stateSchema" : [

    ],
    "availability" : {
      "macOS" : "13.0",
      "tvOS" : "16.0",
      "visionOS" : "1.0",
      "watchOS" : "9.0",
      "iOS" : "16.0",
      "macCatalyst" : "16.0"
    },
    "modelType" : {
      "name" : "MLModelType_mlProgram"
    },
    "userDefinedMetadata" : {
      "snap_gates" : "{\"line\": {\"conf\": 0.3, \"resid\": 0.02}, \"rectangle\": {\"conf\": 0.3, \"resid\": 0.1}, \"triangle\": {\"conf\": 0.3, \"resid\": 1.0}, \"ellipse\": {\"conf\": 0.7, \"resid\": 0.05}, \"star\": {\"conf\": 0.75, \"resid\": 0.08}}",
      "com.github.apple.coremltools.source" : "torch==2.12.1",
      "seq_len" : "256",
      "classes" : "[\"line\", \"rectangle\", \"triangle\", \"ellipse\", \"star\", \"none\"]",
      "com.github.apple.coremltools.conversion_date" : "2026-06-26",
      "com.github.apple.coremltools.version" : "9.0",
      "com.github.apple.coremltools.source_dialect" : "TorchScript",
      "preprocess" : "{\"spacing\": 0.02, \"dist_mean\": 0.019319261383684345, \"dist_std\": 0.0019667792003264808, \"add_curvature\": false}"
    },
    "inputSchema" : [
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1 × 256 × 3)",
        "shortDescription" : "",
        "shape" : "[1, 256, 3]",
        "name" : "features",
        "type" : "MultiArray"
      },
      {
        "hasShapeFlexibility" : "0",
        "isOptional" : "0",
        "dataType" : "Float32",
        "formattedType" : "MultiArray (Float32 1 × 256)",
        "shortDescription" : "",
        "shape" : "[1, 256]",
        "name" : "mask",
        "type" : "MultiArray"
      }
    ],
    "generatedClassName" : "shapes",
    "method" : "predict"
  }
]