Update modeling_nula.py
Browse files- modeling_nula.py +2 -3
modeling_nula.py
CHANGED
|
@@ -3,8 +3,7 @@ import torch.nn as nn
|
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from transformers import PreTrainedModel
|
| 5 |
from transformers.modeling_outputs import ImageClassifierOutput
|
| 6 |
-
from .configuration_nula import
|
| 7 |
-
|
| 8 |
|
| 9 |
class BlurPool2d(nn.Module):
|
| 10 |
def __init__(self, channels, stride=2):
|
|
@@ -100,7 +99,7 @@ class NULAConvBlock(nn.Module):
|
|
| 100 |
return out
|
| 101 |
|
| 102 |
class NulaForImageClassification(PreTrainedModel):
|
| 103 |
-
config_class =
|
| 104 |
base_model_prefix = "nula"
|
| 105 |
|
| 106 |
def __init__(self, config):
|
|
|
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from transformers import PreTrainedModel
|
| 5 |
from transformers.modeling_outputs import ImageClassifierOutput
|
| 6 |
+
from .configuration_nula import NulaConfig
|
|
|
|
| 7 |
|
| 8 |
class BlurPool2d(nn.Module):
|
| 9 |
def __init__(self, channels, stride=2):
|
|
|
|
| 99 |
return out
|
| 100 |
|
| 101 |
class NulaForImageClassification(PreTrainedModel):
|
| 102 |
+
config_class = NulaConfig
|
| 103 |
base_model_prefix = "nula"
|
| 104 |
|
| 105 |
def __init__(self, config):
|