Weilong30 commited on
Commit
56bb42a
1 Parent(s): 88cbd5d

Update modelling_RW.py

Browse files
Files changed (1) hide show
  1. modelling_RW.py +1 -1
modelling_RW.py CHANGED
@@ -959,7 +959,7 @@ class RWForTokenClassification(RWPreTrainedModel):
959
  else:
960
  classifier_dropout = 0.1
961
  self.dropout = nn.Dropout(classifier_dropout)
962
- self.classifier = nn.Linear(config.hidden_size, config.num_lab els)
963
 
964
  # Initialize weights and apply final processing
965
  self.post_init()
 
959
  else:
960
  classifier_dropout = 0.1
961
  self.dropout = nn.Dropout(classifier_dropout)
962
+ self.classifier = nn.Linear(config.hidden_size, config.num_labels)
963
 
964
  # Initialize weights and apply final processing
965
  self.post_init()