Methodbuild_optimizer(self, trn, lr, epochs, gradient_accumulation, warmup_steps, weight_decay, **kwargs)
hanlp/components/amr/seq2seq/seq2seq_amr_parser.py:119
Methodbuild_optimizer(self, optimizer, learning_rate, epsilon, weight_decay_rate, clipnorm, use_amp, train_steps,
hanlp/components/taggers/transformers/transformer_tagger_tf.py:58
Methodbuild_optimizer(self, optimizer, learning_rate, epsilon, weight_decay_rate, clipnorm, fp16, train_steps,
hanlp/components/parsers/biaffine_parser_tf.py:471
Methodbuild_optimizer(self, optimizer, use_amp, train_steps, warmup_steps, **kwargs)
hanlp/components/classifiers/transformer_classifier_tf.py:159
Methodbuild_optimizer(self, trn, epochs, gradient_accumulation=1, lr=1e-3, transformer_lr=5e-5, adam_epsilon=1e-8,
hanlp/components/sts/transformer_sts.py:64
Methodcall(self, inputs, sequence_lengths=None, mask=None, training=None, **kwargs)
hanlp/layers/crf/crf_layer_tf.py:88
Methodcollect_outputs_extend(self, predictions, arc_preds, rel_preds, lens, mask)
hanlp/components/parsers/biaffine/biaffine_sdp.py:134
Methodcollect_outputs_extend(self, predictions: list, arc_preds, rel_preds, lens, mask)
hanlp/components/parsers/biaffine/biaffine_2nd_dep.py:197
Methodcompute_loss(self, arc_scores, rel_scores, arcs, rels, mask, criterion, batch=None)
hanlp/components/parsers/biaffine/structual_attention.py:187
Methodcompute_loss(self, arc_scores, rel_scores, arcs, rels, mask: torch.BoolTensor, criterion, batch=None)
hanlp/components/parsers/biaffine/biaffine_sdp.py:64
Methodcompute_loss(self, arc_scores, rel_scores, arcs, rels, mask, criterion, batch=None)
hanlp/components/parsers/biaffine/biaffine_2nd_dep.py:125
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/sdp.py:99
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/dep_2nd.py:83
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/pos.py:134
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/constituency.py:112
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/__init__.py:109
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/dep.py:98
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/amr.py:94
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/ud.py:92
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/lem.py:97
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/ner/biaffine_ner.py:69
Methodcompute_loss(self,
batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str,
hanlp/components/mtl/tasks/ner/tag_ner.py:135
Methodcompute_loss(self, batch: Dict[str, Any],
output: Union[torch.Tensor, Dict[str, torch.Tensor], Iterab
hanlp/components/mtl/tasks/srl/bio_srl.py:79