File size: 437 Bytes
70916d6 |
1 2 3 4 5 6 7 8 9 10 11 12 |
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) |