Update modeling_nula.py
Browse files- modeling_nula.py +2 -2
modeling_nula.py
CHANGED
|
@@ -1,4 +1,4 @@
|
|
| 1 |
-
import torch
|
| 2 |
import torch.nn as nn
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from transformers import PreTrainedModel
|
|
@@ -99,7 +99,7 @@ class NULAConvBlock(nn.Module):
|
|
| 99 |
out = self.out_act(out)
|
| 100 |
return out
|
| 101 |
|
| 102 |
-
class
|
| 103 |
config_class = NULAConfig
|
| 104 |
base_model_prefix = "nula"
|
| 105 |
|
|
|
|
| 1 |
+
import torch as pt
|
| 2 |
import torch.nn as nn
|
| 3 |
import torch.nn.functional as F
|
| 4 |
from transformers import PreTrainedModel
|
|
|
|
| 99 |
out = self.out_act(out)
|
| 100 |
return out
|
| 101 |
|
| 102 |
+
class NulaForImageClassification(PreTrainedModel):
|
| 103 |
config_class = NULAConfig
|
| 104 |
base_model_prefix = "nula"
|
| 105 |
|