_cm.json

{
  "alias": "app-image-classification-torch-py",
  "automation_alias": "script",
  "automation_uid": "5b4e0237da074764",
  "category": "Modular ML/AI applications",
  "default_env": {
    "CM_BATCH_COUNT": "1",
    "CM_BATCH_SIZE": "1"
  },
  "deps": [
    {
      "tags": "detect,os"
    },
    {
      "tags": "get,sys-utils-cm"
    },
    {
      "names": [
        "python",
        "python3"
      ],
      "tags": "get,python3"
    },
    {
      "tags": "get,dataset,imagenet,image-classification,preprocessed"
    },
    {
      "tags": "get,dataset-aux,imagenet-aux,image-classification"
    },
    {
      "tags": "get,imagenet-helper"
    },
    {
      "tags": "get,ml-model,image-classification,resnet50,_pytorch,_fp32"
    },
    {
      "tags": "get,generic-python-lib,_torch",
      "skip_if_env": {
        "USE_CUDA": [
          "yes"
        ]
      }
    },
    {
      "tags": "get,generic-python-lib,_torch_cuda",
      "enable_if_env": {
        "USE_CUDA": [
          "yes"
        ]
      }
    },
    {
      "tags": "get,generic-python-lib,_torchvision",
      "skip_if_env": {
        "USE_CUDA": [
          "yes"
        ]
      }
    },
    {
      "tags": "get,generic-python-lib,_torchvision_cuda",
      "enable_if_env": {
        "USE_CUDA": [
          "yes"
        ]
      }
    }
  ],
  "tags": [
    "app",
    "image-classification",
    "torch",
    "python"
  ],
  "variations": {
    "cuda": {
      "env": {
         "USE_CUDA": "yes"
      },
      "deps": [
        {
          "tags": "get,cuda"
        }
      ]
    }
  },
  "uid": "e3986ae887b84ca8"
}