| { | |
| "architectures": [ | |
| "CXRModel" | |
| ], | |
| "auto_map": { | |
| "AutoConfig": "configuration.CXRConfig", | |
| "AutoModel": "modeling.CXRModel" | |
| }, | |
| "backbone": "tf_efficientnetv2_s", | |
| "cls_dropout": 0.1, | |
| "cls_num_classes": 5, | |
| "decoder_attention_type": null, | |
| "decoder_center_block": false, | |
| "decoder_channels": [ | |
| 256, | |
| 128, | |
| 64, | |
| 32, | |
| 16 | |
| ], | |
| "decoder_n_blocks": 5, | |
| "decoder_norm_layer": "bn", | |
| "encoder_channels": [ | |
| 24, | |
| 48, | |
| 64, | |
| 160, | |
| 256 | |
| ], | |
| "feature_dim": 256, | |
| "img_size": [ | |
| 320, | |
| 320 | |
| ], | |
| "in_chans": 1, | |
| "model_type": "cxr_basic", | |
| "seg_dropout": 0.1, | |
| "seg_num_classes": 4, | |
| "torch_dtype": "float32", | |
| "transformers_version": "4.47.0" | |
| } | |