File size: 423 Bytes
fc8cd9c |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 |
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) |