test_cls / configuration_mobilenetv3_large_100.py
imliuhan's picture
Upload configuration_mobilenetv3_large_100.py with huggingface_hub
fc8cd9c
raw
history blame contribute delete
423 Bytes
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)