FredZhang7 commited on
Commit
3c3ad8c
·
1 Parent(s): c0951b1

Update modeling_customefficientnetv2.py

Browse files
Files changed (1) hide show
  1. modeling_customefficientnetv2.py +2 -2
modeling_customefficientnetv2.py CHANGED
@@ -1,9 +1,9 @@
1
  from transformers import PretrainedModel
2
- from configuration_efficientnetv2_5 import EfficientNetV2_5Config
3
  import torch
4
 
5
  class CustomEfficientNetV2(PretrainedModel):
6
- config_class = EfficientNetV2_5Config
7
 
8
  def __init__(self, config):
9
  super().__init__(config)
 
1
  from transformers import PretrainedModel
2
+ from configuration_customefficientnetv2 import CustomEfficientNetV2Config
3
  import torch
4
 
5
  class CustomEfficientNetV2(PretrainedModel):
6
+ config_class = CustomEfficientNetV2Config
7
 
8
  def __init__(self, config):
9
  super().__init__(config)