from transformers import PretrainedConfig

class ResnetFeatureExtractorConfig(PretrainedConfig):
    model_type = "resnet"

    def __init__(self, name = 'resnet152', **kwargs):
        if name != 'resnet152':
            raise ValueError(f"`name` must be 'resnet152', got {name}.")

        self.name = name
        super().__init__(**kwargs)