| from transformers import PretrainedConfig | |
| class BoneAgeConfig(PretrainedConfig): | |
| model_type = "bone_age" | |
| def __init__( | |
| self, | |
| backbone="convnextv2_tiny", | |
| feature_dim=768, | |
| dropout=0.1, | |
| num_classes=240, | |
| in_chans=2, | |
| num_models=3, | |
| **kwargs | |
| ): | |
| self.backbone = backbone | |
| self.feature_dim = feature_dim | |
| self.dropout = dropout | |
| self.num_classes = num_classes | |
| self.in_chans = in_chans | |
| self.num_models = num_models | |
| super().__init__(**kwargs) | |