↓ 2 callersMethod__init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-6,
weight_decay=0, adam=False)
language/bert/bert_code/fairseq/optim/lamb.py:93
↓ 2 callersMethod__init__(self, params, lr=1e-3, betas=(0.9, 0.999), eps=1e-8,
weight_decay=0, amsgrad=False)
language/bert/bert_code/fairseq/optim/adam.py:102
↓ 2 callersFunction_apply_del_words(
in_tokens, in_scores, in_attn, word_del_pred, padding_idx, bos_idx, eos_idx
)
language/bert/bert_code/fairseq/models/levenshtein_transformer.py:196
↓ 2 callersFunction_apply_ins_masks(
in_tokens, in_scores, mask_ins_pred, padding_idx, unk_idx, eos_idx
)
language/bert/bert_code/fairseq/models/levenshtein_transformer.py:145
↓ 2 callersMethod_decode_one(
self, tokens, model, encoder_out, incremental_states, log_probs,
temperature=1.,
)
language/bert/bert_code/fairseq/sequence_generator.py:576
↓ 2 callersMethod_generate(
self,
model,
sample,
prefix_tokens=None,
bos_token=None,
**k
language/bert/bert_code/fairseq/sequence_generator.py:116