from transformers import PretrainedConfig class ResnetConfig(PretrainedConfig): model_type = "ResNet50AffectiveFeatureExtractor" def __init__( self, **kwargs, ): super().__init__(**kwargs)