| { |
| "format_version": "rev-safetensors-1", |
| "schema_version": "rev-1.0", |
| "labels": [ |
| "Medical material", |
| "Arterial wall calcification", |
| "Cardiomegaly", |
| "Pericardial effusion", |
| "Coronary artery wall calcification", |
| "Hiatal hernia", |
| "Lymphadenopathy", |
| "Emphysema", |
| "Atelectasis", |
| "Lung nodule", |
| "Lung opacity", |
| "Pulmonary fibrotic sequela", |
| "Pleural effusion", |
| "Mosaic attenuation pattern", |
| "Peribronchial thickening", |
| "Consolidation", |
| "Bronchiectasis", |
| "Interlobular septal thickening" |
| ], |
| "feature_dim": 768, |
| "encoder": { |
| "repo_id": "microsoft/colipri", |
| "revision": "892e42ec363f6c561b89c459d60fcd95ac1a99bf", |
| "package_version": "0.1.3", |
| "pool": true, |
| "project": true, |
| "normalize": false, |
| "input_size": [ |
| 192, |
| 192, |
| 192 |
| ], |
| "dtype": "float32", |
| "feature_storage_dtype": "float16" |
| }, |
| "source": { |
| "repo_id": "chrisvoncsefalvay/ct-rate-colipri-probe", |
| "revision": "e93a9ecd27e4153bab56258bdd97f7c4668ff9e2", |
| "artifact": "typed_calibration.joblib" |
| }, |
| "calibration_input": "raw_probability", |
| "default_head_mode": "reference", |
| "model_type": "decision-model", |
| "architectures": [ |
| "RevModel" |
| ], |
| "tensor_specs": { |
| "head.decision_bias": { |
| "shape": [ |
| 18 |
| ], |
| "dtype": "float64" |
| }, |
| "head.raw_bias": { |
| "shape": [ |
| 18 |
| ], |
| "dtype": "float64" |
| }, |
| "head.weight": { |
| "shape": [ |
| 18, |
| 768 |
| ], |
| "dtype": "float64" |
| }, |
| "scaler.mean": { |
| "shape": [ |
| 768 |
| ], |
| "dtype": "float64" |
| }, |
| "scaler.scale": { |
| "shape": [ |
| 768 |
| ], |
| "dtype": "float64" |
| }, |
| "calibration.0.x": { |
| "shape": [ |
| 128 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.0.y": { |
| "shape": [ |
| 128 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.1.x": { |
| "shape": [ |
| 148 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.1.y": { |
| "shape": [ |
| 148 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.10.x": { |
| "shape": [ |
| 168 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.10.y": { |
| "shape": [ |
| 168 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.11.x": { |
| "shape": [ |
| 158 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.11.y": { |
| "shape": [ |
| 158 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.12.x": { |
| "shape": [ |
| 106 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.12.y": { |
| "shape": [ |
| 106 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.13.x": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.13.y": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.14.x": { |
| "shape": [ |
| 132 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.14.y": { |
| "shape": [ |
| 132 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.15.x": { |
| "shape": [ |
| 134 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.15.y": { |
| "shape": [ |
| 134 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.16.x": { |
| "shape": [ |
| 120 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.16.y": { |
| "shape": [ |
| 120 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.17.x": { |
| "shape": [ |
| 112 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.17.y": { |
| "shape": [ |
| 112 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.2.x": { |
| "shape": [ |
| 128 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.2.y": { |
| "shape": [ |
| 128 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.3.x": { |
| "shape": [ |
| 122 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.3.y": { |
| "shape": [ |
| 122 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.4.x": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.4.y": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.5.x": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.5.y": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.6.x": { |
| "shape": [ |
| 136 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.6.y": { |
| "shape": [ |
| 136 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.7.x": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.7.y": { |
| "shape": [ |
| 138 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.8.x": { |
| "shape": [ |
| 162 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.8.y": { |
| "shape": [ |
| 162 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.9.x": { |
| "shape": [ |
| 158 |
| ], |
| "dtype": "float32" |
| }, |
| "calibration.9.y": { |
| "shape": [ |
| 158 |
| ], |
| "dtype": "float32" |
| }, |
| "decision.threshold_bias": { |
| "shape": [ |
| 18 |
| ], |
| "dtype": "float32" |
| }, |
| "head.original_bias": { |
| "shape": [ |
| 18 |
| ], |
| "dtype": "float32" |
| }, |
| "head.original_weight": { |
| "shape": [ |
| 18, |
| 768 |
| ], |
| "dtype": "float32" |
| } |
| }, |
| "reference_dtype": "float32", |
| "folded_dtype": "float64", |
| "encoder_bundled": true, |
| "encoder_model_config": { |
| "image_encoder": { |
| "backbone": { |
| "_target_": "dynamic_network_architectures.architectures.primus.Primus", |
| "input_channels": 1, |
| "num_classes": 1, |
| "eva_depth": 16, |
| "eva_numheads": 12, |
| "embed_dim": 864, |
| "patch_embed_size": [ |
| 8, |
| 8, |
| 8 |
| ], |
| "input_shape": [ |
| 192, |
| 192, |
| 192 |
| ], |
| "use_rot_pos_emb": true, |
| "use_abs_pos_embed": false, |
| "drop_path_rate": 0.2, |
| "init_values": 0.1, |
| "scale_attn_inner": true, |
| "num_register_tokens": 0 |
| }, |
| "pooler": { |
| "_target_": "colipri.pooling.AttentionPool1D", |
| "embed_dim": 768, |
| "num_heads": 12 |
| }, |
| "projector": { |
| "_target_": "torch.nn.Conv3d", |
| "in_channels": 864, |
| "out_channels": 768, |
| "kernel_size": 1 |
| }, |
| "_target_": "colipri.model.image.ImageEncoder" |
| }, |
| "text_encoder": null, |
| "_target_": "colipri.model.multimodal.Model", |
| "temperature": 1.0 |
| }, |
| "dtype": "float32" |
| } |
|
|
