Update modeling_gptbert.py
Browse files- modeling_gptbert.py +4 -0
modeling_gptbert.py
CHANGED
|
@@ -864,6 +864,10 @@ class Classifier(nn.Module):
|
|
| 864 |
# self.emb2vocab = CastedLinear(config.hidden_size, num_labels, bias=True)
|
| 865 |
self.dropout = nn.Dropout(drop_out)
|
| 866 |
|
|
|
|
|
|
|
|
|
|
|
|
|
| 867 |
self.initialize(config.hidden_size, config.intermediate_size, num_labels)
|
| 868 |
|
| 869 |
@torch.no_grad()
|
|
|
|
| 864 |
# self.emb2vocab = CastedLinear(config.hidden_size, num_labels, bias=True)
|
| 865 |
self.dropout = nn.Dropout(drop_out)
|
| 866 |
|
| 867 |
+
print(f"Hidden Size: {config.hidden_size}")
|
| 868 |
+
print(f"Intermediate Size: {config.intermediate_size}")
|
| 869 |
+
print(f"Proj_std: {math.sqrt(2.0 / (hidden_size + intermediate_size))}")
|
| 870 |
+
|
| 871 |
self.initialize(config.hidden_size, config.intermediate_size, num_labels)
|
| 872 |
|
| 873 |
@torch.no_grad()
|