Upload model
Browse files- configuration.py +2 -0
- modeling_fsg.py +2 -0
configuration.py
CHANGED
|
@@ -43,6 +43,8 @@ class CompareNetConfig:
|
|
| 43 |
|
| 44 |
|
| 45 |
class FsgConfig(PretrainedConfig):
|
|
|
|
|
|
|
| 46 |
def __init__(
|
| 47 |
self,
|
| 48 |
fe_config=None,
|
|
|
|
| 43 |
|
| 44 |
|
| 45 |
class FsgConfig(PretrainedConfig):
|
| 46 |
+
model_type = "fsg"
|
| 47 |
+
|
| 48 |
def __init__(
|
| 49 |
self,
|
| 50 |
fe_config=None,
|
modeling_fsg.py
CHANGED
|
@@ -383,6 +383,8 @@ class FsgModel(
|
|
| 383 |
plt.show()
|
| 384 |
```
|
| 385 |
"""
|
|
|
|
|
|
|
| 386 |
|
| 387 |
def __init__(self, config: FsgConfig, **kwargs):
|
| 388 |
super().__init__(config)
|
|
|
|
| 383 |
plt.show()
|
| 384 |
```
|
| 385 |
"""
|
| 386 |
+
config_class = FsgConfig
|
| 387 |
+
|
| 388 |
|
| 389 |
def __init__(self, config: FsgConfig, **kwargs):
|
| 390 |
super().__init__(config)
|