Upload model
Browse files- prot_albert_model.py +0 -4
prot_albert_model.py
CHANGED
|
@@ -24,10 +24,6 @@ class ProtAlBertModel(PreTrainedModel):
|
|
| 24 |
self.protbert = AlbertModel.from_pretrained(
|
| 25 |
model_name, trust_remote_code=True, config=self.config
|
| 26 |
)
|
| 27 |
-
print(self.config)
|
| 28 |
-
print(f"hidden_size: {hidden_size}")
|
| 29 |
-
print(f"num_labels: {num_labels}")
|
| 30 |
-
print(f"CONFIG HIDDEN : {self.config.hidden_size}")
|
| 31 |
num_labels = 10 if type(num_labels) is not int else num_labels
|
| 32 |
self.last_layer = nn.Sequential(
|
| 33 |
nn.Dropout(0.1),
|
|
|
|
| 24 |
self.protbert = AlbertModel.from_pretrained(
|
| 25 |
model_name, trust_remote_code=True, config=self.config
|
| 26 |
)
|
|
|
|
|
|
|
|
|
|
|
|
|
| 27 |
num_labels = 10 if type(num_labels) is not int else num_labels
|
| 28 |
self.last_layer = nn.Sequential(
|
| 29 |
nn.Dropout(0.1),
|