{
  "arch": "resnet18",
  "num_classes": 4,
  "dropout_p": 0.5,
  "pretrained": false,
  "model_type": "baseline"
}