Image Classification
Transformers
PyTorch
efficientnetv25
custom_code
FredZhang7 commited on
Commit
ea15e8c
·
1 Parent(s): 5f51f01

Update configuration_customefficientnetv2.py

Browse files
configuration_customefficientnetv2.py CHANGED
@@ -1,6 +1,6 @@
1
  from transformers import PretrainedConfig
2
 
3
- class EfficientNetV2_5Config(PretrainedConfig):
4
  model_type = "EfficientNetV2_5"
5
  keys_to_ignore_at_inference = ["efficientnetv2"]
6
 
 
1
  from transformers import PretrainedConfig
2
 
3
+ class CustomEfficientNetV2Config(PretrainedConfig):
4
  model_type = "EfficientNetV2_5"
5
  keys_to_ignore_at_inference = ["efficientnetv2"]
6