{
  "check_exit_status": "yes",
  "customize": {
    "install_env": {
      "MODEL_NAME": "profile",
      "PACKAGE_VERSION": "1",
      "QAIC_TOOLCHAIN_PATH": "/opt/qti-aic/",
      "_BATCH_SIZE": "1"
    },
    "no_os_in_suggested_path": "yes",
    "no_ver_in_suggested_path": "yes",
    "package_name": "Profile for QAIC model",
    "skip_file_check": "yes"
  },
  "deps": {
    "coco-cal": {
      "enable_if_env": [
        {
          "CK_CALIBRATE_COCO": [
            "on",
            "ON",
            "yes",
            "YES",
            "true",
            "TRUE",
            "1"
          ]
        }
      ],
      "local": "yes",
      "name": "COCO calibration dataset in NCHW",
      "sort": 100,
      "tags": "dataset,calibration,coco.2017,for-ssd-resnet-onnx-preprocessed",
      "update_tags_if_env": {
        "first.5": [
          {
            "_CAL": "calibrate_first.5"
          }
        ],
        "mlperf": [
          {
            "_COCO_CAL": "calibrate_ssd_resnet34_mlperf"
          }
        ]
      }
    },
    "imagenet-cal": {
      "enable_if_env": [
        {
          "CK_CALIBRATE_IMAGENET": [
            "on",
            "ON",
            "yes",
            "YES",
            "true",
            "TRUE",
            "1"
          ]
        }
      ],
      "local": "yes",
      "name": "ImageNet calibration dataset",
      "sort": 100,
      "tags": "dataset,imagenet,calibration,preprocessed,for.resnet",
      "update_tags_if_env": {
        "first.5": [
          {
            "_CAL": "calibrate_first.5"
          }
        ],
        "layout.nhwc": [
          {
            "_MODEL_TO_CONVERT": "calibrate_resnet50_tf"
          }
        ],
        "mlperf.option1": [
          {
            "_IMAGENET_CAL": "calibrate_mlperf.option1"
          }
        ],
        "mlperf.option2": [
          {
            "_IMAGENET_CAL": "calibrate_mlperf.option2"
          }
        ]
      }
    },
    "model-source": {
      "local": "yes",
      "name": "Original model",
      "sort": 5,
      "tags": "model",
      "update_tags_if_env": {
        "resnet50,tf,mlperf,fix_input_shape": [
          {
            "_MODEL_TO_CONVERT": "calibrate_resnet50_tf"
          }
        ],
        "ssd-resnet34,onnx,mlperf,non-quantized,no-nms": [
          {
            "_MODEL_TO_CONVERT": "calibrate_ssd_resnet34_no_nms_mlperf"
          }
        ]
      }
    }
  },
  "end_full_path_universal": "profile.yaml",
  "process_script": "install",
  "soft_uoa": "compiler.glow.profile",
  "suggested_path": "model-profile-qaic",
  "tags": [
    "profile",
    "model-profile",
    "qaic",
    "qaic-profile"
  ],
  "variations": {
    "bs.1": {
      "extra_env": {
        "_BATCH_SIZE": "1"
      }
    },
    "bs.8": {
      "extra_env": {
        "_BATCH_SIZE": "8"
      }
    },
    "first.5": {
      "extra_env": {
        "_CAL": "calibrate_first.5"
      }
    },
    "mlperf.option1": {
      "extra_env": {
        "_IMAGENET_CAL": "calibrate_mlperf.option1"
      }
    },
    "mlperf.option2": {
      "extra_env": {
        "_IMAGENET_CAL": "calibrate_mlperf.option2"
      }
    },
    "resnet50": {
      "extra_customize": {
        "package_name": "Profile for QAIC model (MLPerf ResNet50 TF)"
      },
      "extra_env": {
        "CK_CALIBRATE_IMAGENET": "yes",
        "CK_ENV_DATASET_PREPROCESSED_DIR": "$<<CK_ENV_DATASET_IMAGENET_PREPROCESSED_DIR>>$",
        "CK_ENV_DATASET_PREPROCESSED_FOF": "$<<CK_ENV_DATASET_IMAGENET_PREPROCESSED_SUBSET_FOF>>$",
        "_COMPILER_ARGS": "",
        "_COMPILER_PARAMS": "-quantization-schema=symmetric_with_uint8 -quantization-precision=Int8 -output-node-name=ArgMax",
        "_MODEL_TO_CONVERT": "calibrate_resnet50_tf",
        "_OUTPUT_NODE_NAME": "-output-node-name=ArgMax"
      },
      "extra_tags": [
        "image-classification",
        "calibrated-by-qaic"
      ]
    },
    "ssd_resnet34": {
      "extra_customize": {
        "package_name": "Profile for QAIC model (MLPerf SSD-ResNet34 No NMS)"
      },
      "extra_env": {
        "CK_CALIBRATE_COCO": "yes",
        "CK_ENV_DATASET_PREPROCESSED_DIR": "$<<CK_ENV_DATASET_OBJ_DETECTION_PREPROCESSED_DIR>>$",
        "CK_ENV_DATASET_PREPROCESSED_FOF": "$<<CK_ENV_DATASET_OBJ_DETECTION_PREPROCESSED_SUBSET_FOF>>$",
        "ML_MODEL_MODEL_NAME": "ssd-resnet34",
        "_COMPILER_ARGS": "",
        "_COMPILER_PARAMS": "",
        "_MODEL_TO_CONVERT": "calibrate_ssd_resnet34_no_nms_mlperf"
      },
      "extra_tags": [
        "object-detection",
        "calibrated-by-qaic",
        "no-nms"
      ]
    }
  }
}