config.json

{
  "architectures": [
    "CrossedOutClassifier"
  ],
  "model_type": "crossed_out_classifier",
  "num_classes": 2,
  "class_names": [
    "no",
    "yes"
  ],
  "image_size": [
    224,
    224
  ],
  "num_channels": 3,
  "backbone": "resnet18",
  "pretrained": true,
  "validation_accuracy": 0.96875,
  "total_parameters": 11187158,
  "framework": "pytorch",
  "task": "image-classification"
}