File size: 326 Bytes
9207b26
 
 
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
{
  "sdk": "docker",
  "task": "age_group_classification",
  "model_name": "mobilenetv3_large_100",
  "num_classes": 4,
  "class_names": ["A", "B", "C", "D"],
  "input_size": [224, 224],
  "normalize": {
    "mean": [0.485, 0.456, 0.406],
    "std":  [0.229, 0.224, 0.225]
  },
  "checkpoint_format": "dict_with_state_dict"
}