mkjaramillo
/

breastCancer

Image Classification

Model card Files Files and versions

mkjaramillo commited on Jun 12, 2023

Commit

aa30f21

·

1 Parent(s): f85095d

Update config.json

Files changed (1) hide show

config.json +2 -2

config.json CHANGED Viewed

@@ -1,6 +1,6 @@
 {
   "architectures": [
-    "ViTForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
   "encoder_stride": 16,
@@ -19,7 +19,7 @@
     "No_Cancer": "1"
   },
   "layer_norm_eps": 1e-12,
-  "model_type": "vit",
   "num_attention_heads": 12,
   "num_channels": 3,
   "num_hidden_layers": 12,

 {
   "architectures": [
+    "ResNetForImageClassification"
   ],
   "attention_probs_dropout_prob": 0.0,
   "encoder_stride": 16,
     "No_Cancer": "1"
   },
   "layer_norm_eps": 1e-12,
+  "model_type": "resnet",
   "num_attention_heads": 12,
   "num_channels": 3,
   "num_hidden_layers": 12,