lgcharpe commited on
Commit
91e4060
·
verified ·
1 Parent(s): a36b0d0

Update modeling_gptbert.py

Browse files
Files changed (1) hide show
  1. 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()