MCPcopy Create free account
hub / github.com/HA-Transformer/MAT / generate

Method generate

fairseq/sequence_generator.py:101–516  ·  view source on GitHub ↗

Generate a batch of translations. Args: models (List[~fairseq.models.FairseqModel]): ensemble of models sample (dict): batch prefix_tokens (torch.LongTensor, optional): force decoder to begin with these tokens

(
        self,
        models,
        sample,
        prefix_tokens=None,
        bos_token=None,
        **kwargs
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 9

test_with_normalizationMethod · 0.95
test_without_normalizationMethod · 0.95
test_with_lenpen_favoring_short_hyposMethod · 0.95
test_with_lenpen_favoring_long_hyposMethod · 0.95
test_maxlenMethod · 0.95
test_diverse_beam_searchMethod · 0.95
test_topp_sampling_search_low_probMethod · 0.95
test_topp_sampling_search_high_probMethod · 0.95
_backtranslation_dataset_helperMethod · 0.95

Calls 10

max_decoder_positionsMethod · 0.95
forward_encoderMethod · 0.95
reorder_encoder_outMethod · 0.95
reorder_incremental_stateMethod · 0.95
forward_decoderMethod · 0.95
EnsembleModelClass · 0.85
set_src_lengthsMethod · 0.80
sizeMethod · 0.45
stepMethod · 0.45
addMethod · 0.45

Tested by 9

test_with_normalizationMethod · 0.76
test_without_normalizationMethod · 0.76
test_with_lenpen_favoring_short_hyposMethod · 0.76
test_with_lenpen_favoring_long_hyposMethod · 0.76
test_maxlenMethod · 0.76
test_diverse_beam_searchMethod · 0.76
test_topp_sampling_search_low_probMethod · 0.76
test_topp_sampling_search_high_probMethod · 0.76
_backtranslation_dataset_helperMethod · 0.76