Commit ·
3c3ad8c
1
Parent(s): c0951b1
Update modeling_customefficientnetv2.py
Browse files
modeling_customefficientnetv2.py
CHANGED
|
@@ -1,9 +1,9 @@
|
|
| 1 |
from transformers import PretrainedModel
|
| 2 |
-
from
|
| 3 |
import torch
|
| 4 |
|
| 5 |
class CustomEfficientNetV2(PretrainedModel):
|
| 6 |
-
config_class =
|
| 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)
|