Update modeling_minGRU.py
Browse files- modeling_minGRU.py +2 -3
modeling_minGRU.py
CHANGED
@@ -95,11 +95,9 @@ class MinGRUForSequenceClassification(PreTrainedModel):
|
|
95 |
logits=logits,
|
96 |
)
|
97 |
|
|
|
98 |
@classmethod
|
99 |
def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
|
100 |
-
"""
|
101 |
-
Load model from a pretrained checkpoint.
|
102 |
-
"""
|
103 |
model = super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
104 |
|
105 |
for name, param in model.named_parameters():
|
@@ -109,6 +107,7 @@ class MinGRUForSequenceClassification(PreTrainedModel):
|
|
109 |
print(f"Initialized parameter {name} manually.")
|
110 |
|
111 |
return model
|
|
|
112 |
|
113 |
def save_pretrained(self, save_directory, safe_serialization: Optional[bool] = True, **kwargs):
|
114 |
"""
|
|
|
95 |
logits=logits,
|
96 |
)
|
97 |
|
98 |
+
"""
|
99 |
@classmethod
|
100 |
def from_pretrained(cls, pretrained_model_name_or_path, *model_args, **kwargs):
|
|
|
|
|
|
|
101 |
model = super().from_pretrained(pretrained_model_name_or_path, *model_args, **kwargs)
|
102 |
|
103 |
for name, param in model.named_parameters():
|
|
|
107 |
print(f"Initialized parameter {name} manually.")
|
108 |
|
109 |
return model
|
110 |
+
"""
|
111 |
|
112 |
def save_pretrained(self, save_directory, safe_serialization: Optional[bool] = True, **kwargs):
|
113 |
"""
|