from transformers import PretrainedConfig | |
class UniformerXXSFinetuneConfig(PretrainedConfig): | |
model_type = "uniformer_finetuned" | |
def __init__( | |
self, | |
pretrained: str = 'uniformer_xxs_400', | |
out_class: int = 20, | |
**kwargs | |
): | |
self.pretrained = pretrained | |
self.out_class = out_class | |
super().__init__(**kwargs) | |