| from transformers import PretrainedConfig | |
| class SwinCXRConfig(PretrainedConfig): | |
| model_type = "swin_cxr" | |
| def __init__(self, num_classes=3, embed_dim=128, num_heads=4, num_layers=4, dropout=0.1, **kwargs): | |
| self.num_classes = num_classes | |
| self.embed_dim = embed_dim | |
| self.num_heads = num_heads | |
| self.num_layers = num_layers | |
| self.dropout = dropout | |
| super().__init__(**kwargs) |