| { | |
| "architectures": [ | |
| "Med3DSeg" | |
| ], | |
| "dice_weight": 1.0, | |
| "embed_dim": 768, | |
| "focal_weight": 1.0, | |
| "freeze_text_encoder": false, | |
| "image_size": [ | |
| 96, | |
| 96, | |
| 96 | |
| ], | |
| "mlp_dim": 3072, | |
| "model_type": "med3d_seg", | |
| "num_heads": 12, | |
| "num_kv_heads": 4, | |
| "pass_num": 1, | |
| "patch_size": [ | |
| 16, | |
| 16, | |
| 16 | |
| ], | |
| "pretrained_model": "", | |
| "text_model": "nlpie/tiny-clinicalbert", | |
| "torch_dtype": "bfloat16", | |
| "transformer_depth": 4, | |
| "transformers_version": "4.52.3", | |
| "wb_project": "Text3DSAM" | |
| } | |