(self)
| 67 | self.freeze() |
| 68 | |
| 69 | def freeze(self): |
| 70 | self.transformer = self.transformer.eval() |
| 71 | #self.train = disabled_train |
| 72 | for param in self.parameters(): |
| 73 | param.requires_grad = False |
| 74 | |
| 75 | def forward(self, text): |
| 76 | batch_encoding = self.tokenizer(text, truncation=True, max_length=self.max_length, return_length=True, |