forked from microsoft/olive-recipes
-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathvit-base-patch16-224_qnn_gpu.json.config
More file actions
116 lines (116 loc) · 3.71 KB
/
vit-base-patch16-224_qnn_gpu.json.config
File metadata and controls
116 lines (116 loc) · 3.71 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
{
"$schema": "https://github.com/microsoft/olive-recipes/raw/refs/heads/main/.aitk/configs/config_schema.json",
"name": "Convert to Qualcomm GPU",
"oliveFile": "QNN/config_gpu_fp32.json",
"runtimeOverwrite": {
"autoGenerated": true,
"executeRequirement": "General/CPU_py3.12.9"
},
"runtime": {
"autoGenerated": true,
"name": "Evaluate on",
"type": "enum",
"displayNames": [
"Qualcomm GPU"
],
"path": "systems.target_system.accelerators.0.execution_providers.0",
"values": [
"QNNExecutionProvider"
],
"readOnly": false
},
"optimizationPaths": [
{
"path": "passes.conversion",
"name": "fp32"
}
],
"optimizationDefault": "fp32",
"sections": [
{
"autoGenerated": true,
"name": "Convert",
"phase": "Conversion",
"parameters": [],
"toggle": {
"autoGenerated": true,
"name": "Convert to ONNX format",
"type": "bool",
"path": "passes.conversion",
"actions": [
[],
[]
],
"readOnly": true
}
},
{
"name": "Evaluate",
"phase": "Evaluation",
"parameters": [
{
"name": "Evaluation Dataset",
"tags": [
"EvaluationDataset"
],
"type": "enum",
"path": "data_configs[0].load_dataset_config.data_name",
"values": [
"timm/mini-imagenet",
"imagenet-1k"
],
"template": {
"path": "data_configs[0].load_dataset_config.data_name",
"values": [
"timm/mini-imagenet",
"imagenet-1k"
],
"template": "EvaluationDataset"
}
},
{
"name": "Evaluation Dataset Split",
"tags": [
"EvaluationDatasetSplit",
"DependsOnDataset"
],
"type": "enum",
"path": "data_configs[0].load_dataset_config.split",
"values": [
"train",
"validation",
"test"
],
"template": {
"path": "data_configs[0].load_dataset_config.split",
"template": "EvaluationDatasetSplit"
}
},
{
"name": "Evaluation Dataset Size",
"type": "int",
"path": "data_configs[0].pre_process_data_config.size",
"template": {
"path": "data_configs[0].pre_process_data_config.size",
"template": "EvaluationDatasetSize"
}
}
],
"toggle": {
"autoGenerated": true,
"name": "Evaluate model performance",
"type": "bool",
"path": "evaluator",
"actions": [
[],
[
{
"type": "delete",
"path": "evaluator"
}
]
]
}
}
]
}