Upload processor

image_processing_efficientnet.py

@@ -13,6 +13,7 @@ class EfficientNetImageProcessor(BaseImageProcessor):
                  ):
         super().__init__(**kwargs)
 
+        self.model_name = model_name
         self.config = resolve_data_config({}, model=create_model(model_name, pretrained=False))
         
 

preprocessor_config.json

@@ -22,5 +22,6 @@
       0.225
     ]
   },
-  "image_processor_type": "EfficientNetImageProcessor"
+  "image_processor_type": "EfficientNetImageProcessor",
+  "model_name": "efficientnet_b0"
 }