P01son Weilong30 commited on
Commit
0b0e7b6
1 Parent(s): 88cbd5d

Update modelling_RW.py (#1)

Browse files

- Update modelling_RW.py (56bb42a05bf4fde9ab8070140d3e6119796caa9e)


Co-authored-by: Weilong Chen <Weilong30@users.noreply.huggingface.co>

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()