Upload configuration_mobilenet.py

configuration_mobilenet.py

@@ -0,0 +1,24 @@
+from transformers import PretrainedConfig
+
+class MobileNetV1Config(PretrainedConfig):
+    model_type = "mobilenet_v1_modern"
+
+    def __init__(
+        self,
+        num_classes: int = 200,
+        block_dropout: float = 0.0,
+        final_dropout: float = 0.50,
+        **kwargs
+    ):
+        self.num_classes = num_classes
+        self.block_dropout = block_dropout
+        self.final_dropout = final_dropout
+        
+        # This mapping allows AutoModel to find the code when pushed to the Hub
+        if "auto_map" not in kwargs:
+            kwargs["auto_map"] = {
+                "AutoConfig": "configuration_mobilenet.MobileNetV1Config",
+                "AutoModelForImageClassification": "modeling_mobilenet.MobileNetV1ForImageClassification"
+            }
+            
+        super().__init__(**kwargs)