davda54 commited on
Commit
3b73353
1 Parent(s): 3586af3

Update modeling_deberta.py

Browse files
Files changed (1) hide show
  1. modeling_deberta.py +15 -0
modeling_deberta.py CHANGED
@@ -1100,6 +1100,8 @@ class DebertaV2ForMaskedLM(DebertaV2PreTrainedModel):
1100
  self.deberta = DebertaV2Model(config)
1101
  self.cls = DebertaV2OnlyMLMHead(config)
1102
 
 
 
1103
  # Initialize weights and apply final processing
1104
  self.post_init()
1105
 
@@ -1130,6 +1132,19 @@ class DebertaV2ForMaskedLM(DebertaV2PreTrainedModel):
1130
 
1131
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
1132
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1133
  outputs = self.deberta(
1134
  input_ids,
1135
  attention_mask=attention_mask,
 
1100
  self.deberta = DebertaV2Model(config)
1101
  self.cls = DebertaV2OnlyMLMHead(config)
1102
 
1103
+ self.verbose = False
1104
+
1105
  # Initialize weights and apply final processing
1106
  self.post_init()
1107
 
 
1132
 
1133
  return_dict = return_dict if return_dict is not None else self.config.use_return_dict
1134
 
1135
+ if self.verbose:
1136
+ for i in input_ids[0, :].tolist():
1137
+ print(i, end=", ")
1138
+ print()
1139
+ if attention_mask is not None:
1140
+ for i in attention_mask[0, :].tolist():
1141
+ print(i, end=", ")
1142
+ print()
1143
+ if position_ids is not None:
1144
+ for i in position_ids[0, :].tolist():
1145
+ print(i, end=", ")
1146
+ print()
1147
+
1148
  outputs = self.deberta(
1149
  input_ids,
1150
  attention_mask=attention_mask,