↓ 3 callersFunctionbuild_adamw_optimizer(config, learning_rate, epsilon, clipnorm, train_steps, use_amp, warmup_steps,
weigh
hanlp/layers/transformers/utils_tf.py:153
↓ 3 callersMethodcompute_loss(self, arc_scores, rel_scores, arcs, rels, mask, criterion, batch=None)
hanlp/components/parsers/biaffine/biaffine_dep.py:487
↓ 3 callersMethoddecode(self, contextualized_embeddings, sent_lengths, masks, gold_arg_starts, gold_arg_ends, gold_arg_labels,
hanlp/components/srl/span_rank/span_ranking_srl_model.py:371
↓ 3 callersMethoddecode_output(self, out, mask, batch, span_probs=None, decoder=None, tokens=None)
hanlp/components/parsers/constituency/crf_constituency_parser.py:118
↓ 3 callersMethodfit(self, trn_data, dev_data, save_dir,
n_embed=100,
pretrained_embed=None,
e
hanlp/components/parsers/biaffine_parser_tf.py:60
↓ 3 callersFunctionrender_labeled_span(b, e, spans, labels, label, offset, unidirectional=False)
plugins/hanlp_common/hanlp_common/visualization.py:180
↓ 3 callersMethodupdate_metrics(self, metric, logits, y, mask, batch=None, prediction=None)
hanlp/components/taggers/tagger.py:110
↓ 2 callersMethod__init__(self, input_size, hidden_size, num_layers=1, bidirectional=False, dropout=0)
hanlp/components/parsers/biaffine/variationalbilstm.py:64
↓ 2 callersMethod__init__(self, hidden_size, n_mlp_arc, n_mlp_rel, mlp_dropout, n_rels, arc_dropout=None,
rel_dropout=
hanlp/components/parsers/biaffine/biaffine_model.py:155