config.json

{
    "model_type": "resnet18",
    "num_labels": 4,
    "labels": ["tom", "jerry", "tom_jerry_0", "tom_jerry_1"]
}