from transformers import PretrainedConfig | |
class RyzenAIORTModelForImageClassificationConfig(PretrainedConfig): | |
# no-format | |
r""" | |
This is the configuration class to store the configuration of a [`RyzenAIORTModelForImageClassification`]. | |
""" | |
model_type = "RyzenAIORTModelForImageClassification_mobilenetv3_large_100" | |
def __init__(self, | |
**kwargs): | |
super().__init__(**kwargs) |