(self, encoder, config, tokenizer, args)
| 139 | |
| 140 | class DefectModel(nn.Module): |
| 141 | def __init__(self, encoder, config, tokenizer, args): |
| 142 | super(DefectModel, self).__init__() |
| 143 | self.encoder = encoder |
| 144 | self.config = config |
| 145 | self.tokenizer = tokenizer |
| 146 | self.classifier = nn.Linear(config.hidden_size, 2) |
| 147 | self.args = args |
| 148 | |
| 149 | def get_t5_vec(self, source_ids): |
| 150 | attention_mask = source_ids.ne(self.tokenizer.pad_token_id) |