Upload model
Browse files- config.json +2 -3
- configuration_efficientnet.py +0 -1
config.json
CHANGED
|
@@ -1,11 +1,10 @@
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
-
"
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_efficientnet.EfficientNetConfig",
|
| 7 |
-
"AutoModel": "modeling_efficientnet.EfficientNetModel"
|
| 8 |
-
"AutoModelForImageClassification": "modeling_efficientnet.EfficientNetModelForImageClassification"
|
| 9 |
},
|
| 10 |
"global_pool": "avg",
|
| 11 |
"model_name": "efficientnet_b0",
|
|
|
|
| 1 |
{
|
| 2 |
"architectures": [
|
| 3 |
+
"EfficientNetModel"
|
| 4 |
],
|
| 5 |
"auto_map": {
|
| 6 |
"AutoConfig": "configuration_efficientnet.EfficientNetConfig",
|
| 7 |
+
"AutoModel": "modeling_efficientnet.EfficientNetModel"
|
|
|
|
| 8 |
},
|
| 9 |
"global_pool": "avg",
|
| 10 |
"model_name": "efficientnet_b0",
|
configuration_efficientnet.py
CHANGED
|
@@ -50,7 +50,6 @@ class EfficientNetOnnxConfig(ViTOnnxConfig):
|
|
| 50 |
|
| 51 |
|
| 52 |
__all__ = [
|
| 53 |
-
'MODEL_NAMES',
|
| 54 |
'EfficientNetConfig',
|
| 55 |
'EfficientNetOnnxConfig'
|
| 56 |
]
|
|
|
|
| 50 |
|
| 51 |
|
| 52 |
__all__ = [
|
|
|
|
| 53 |
'EfficientNetConfig',
|
| 54 |
'EfficientNetOnnxConfig'
|
| 55 |
]
|