Commit ·
55a56ca
1
Parent(s): 4d22bde
Delete configuration_customefficientnetv2.py
Browse files
configuration_customefficientnetv2.py
DELETED
|
@@ -1,15 +0,0 @@
|
|
| 1 |
-
from transformers import PretrainedConfig
|
| 2 |
-
|
| 3 |
-
class CustomEfficientNetV2Config(PretrainedConfig):
|
| 4 |
-
model_type = "EfficientNetV2_5"
|
| 5 |
-
keys_to_ignore_at_inference = ["efficientnetv2"]
|
| 6 |
-
|
| 7 |
-
def __init__(self,
|
| 8 |
-
url,
|
| 9 |
-
input_size,
|
| 10 |
-
num_classes,
|
| 11 |
-
**kwargs):
|
| 12 |
-
self.url = url
|
| 13 |
-
self.input_size = input_size
|
| 14 |
-
self.num_classes = num_classes
|
| 15 |
-
# super().__init__(**kwargs)
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|