Upload config.json with huggingface_hub

config.json

@@ -0,0 +1,82 @@
+{
+  "model_type": "mirai",
+  "architectures": ["MiraiModel"],
+  "auto_map": {
+    "AutoConfig": "configuration_mirai.MiraiConfig",
+    "AutoModel": "modeling_mirai.MiraiModel"
+  },
+  "model_name": "mirai_full",
+  "num_classes": 5,
+  "img_size": [1664, 2048],
+  "num_chan": 3,
+  "num_images": 4,
+  "multi_image": true,
+
+  "encoder_config": {
+    "architecture": "resnet",
+    "block_layout": [
+      ["BasicBlock", 2],
+      ["BasicBlock", 2],
+      ["BasicBlock", 2],
+      ["BasicBlock", 2]
+    ],
+    "pool_name": "RiskFactorPool",
+    "img_only_dim": 512,
+    "hidden_dim": 612,
+    "dropout": 0.25,
+    "pretrained_on_imagenet": false
+  },
+
+  "transformer_config": {
+    "hidden_dim": 512,
+    "num_heads": 8,
+    "num_layers": 1,
+    "dropout": 0.1,
+    "max_seq_length": 4,
+    "output_dim": 612
+  },
+
+  "risk_factors": {
+    "use_risk_factors": true,
+    "num_risk_factors": 34,
+    "risk_factor_keys": [
+      "density", "binary_family_history", "binary_biopsy_benign", "binary_biopsy_LCIS",
+      "binary_biopsy_atypical_hyperplasia", "age", "menarche_age", "menopause_age",
+      "first_pregnancy_age", "prior_hist", "race", "parous", "menopausal_status",
+      "weight", "height", "ovarian_cancer", "ovarian_cancer_age", "ashkenazi",
+      "brca", "mom_bc_cancer_history", "m_aunt_bc_cancer_history",
+      "p_aunt_bc_cancer_history", "m_grandmother_bc_cancer_history",
+      "p_grantmother_bc_cancer_history", "sister_bc_cancer_history",
+      "mom_oc_cancer_history", "m_aunt_oc_cancer_history",
+      "p_aunt_oc_cancer_history", "m_grandmother_oc_cancer_history",
+      "p_grantmother_oc_cancer_history", "sister_oc_cancer_history",
+      "hrt_type", "hrt_duration", "hrt_years_ago_stopped"
+    ]
+  },
+
+  "preprocessing": {
+    "img_mean": 7047.99,
+    "img_std": 12005.5,
+    "normalize_method": "imagenet",
+    "imagenet_mean": [0.485, 0.456, 0.406],
+    "imagenet_std": [0.229, 0.224, 0.225]
+  },
+
+  "training_config": {
+    "batch_size": 1,
+    "learning_rate": 1e-4,
+    "weight_decay": 0.0001,
+    "max_epochs": 100,
+    "early_stopping_patience": 10
+  },
+
+  "inference_config": {
+    "batch_size": 1,
+    "use_gpu": true,
+    "fp16": false
+  },
+
+  "model_version": "1.0.0",
+  "pytorch_version": ">=1.7.0",
+  "transformers_version": ">=4.20.0"
+}